Створення вектора нулів для певного розміру

Я хотів би ініціалізувати вектор нулів із певним розміром, який визначається під час виконання.

У C це буде так:

int main(void)
{
    uint size = get_uchar();
    int A[size][size];
    memset(A, 0, size*size*sizeof(int));
}

Ось допоміжна функція, яку я намагався писати в Rust, але я думаю, що синтаксис нарізки 0..sizeображає компілятор. До того ж, це виглядає більш багатослівно, ніж версія C. Чи є більш ідіоматичний спосіб зробити це?

fn zeros(size: u32) -> Vec<i32> {
    let mut zero_vec: Vec<i32> = Vec::with_capacity(size);
    for i in 0..size {
        zero_vec.push(0);
    }
    return zero_vec;
}

Клянусь, що старі документи використовували тут пояснення from_elem()методу, і, здається, жодна з перестановок нотації не працює[0 ; size]

Я хотів би вкласти це в алгоритм пошуку підрядків:

pub fn kmp(text: &str, pattern: &str) -> i64 {
    let mut shifts = zeros(pattern.len()+1);
}

Для ініціалізації вектора нулів (або будь-якого іншого постійного значення) заданої довжини можна використовувати vec!макрос:

let len = 10;
let zero_vec = vec![0; len];

Тим не менш, ваша функція спрацювала для мене після декількох виправлень синтаксису:

fn zeros(size: u32) -> Vec<i32> {
    let mut zero_vec: Vec<i32> = Vec::with_capacity(size as usize);
    for i in 0..size {
        zero_vec.push(0);
    }
    return zero_vec;
}

uintбільше не існує в Rust 1.0, його sizeпотрібно було відтворити як usize, і типи для векторів, які повинні збігатися (змінено let mut zero_vec: Vec<i64>на let mut zero_vec: Vec<i32>.

Ось інший спосіб, набагато коротший. Він працює з Rust 1.0:

fn zeros(size: u32) -> Vec<i32> {
    vec![0; size as usize]
}

fn main() {
    let vec = zeros(10);
    for i in vec.iter() {
        println!("{}", i)
    }
}

Ви також можете скористатися iter::repeatфункцією, яка, на мою думку, є "більш ідіоматичною" (і мені це просто виглядає приємніше):

use std::iter;

fn zeros(size: usize) -> Vec<i32> {
    iter::repeat(0).take(size).collect()
}

Ви можете використовувати розмір

let mut v = Vec::new();
let l = 42;
v.resize(l, 0u32);