如何用零填充数组？

Question

fn main() {
    let arr: [u8;8] = [97, 112, 112, 108, 101];
    println!("Len is {}",arr.len());
    println!("Elements are {:?}",arr);
}

error[E0308]: mismatched types
 --> src/main.rs:2:23
  |
2 |     let arr: [u8;8] = [97, 112, 112, 108, 101];
  |              ------   ^^^^^^^^^^^^^^^^^^^^^^^^ expected an array with a fixed size of 8 elements, found one with 5 elements
  |              |
  |              expected due to this

有什么方法可以用0填充剩余的元素吗？类似于：

let arr: [u8;8] = [97, 112, 112, 108, 101].something();

Answer 1

您可以使用 concat_arrays 宏：

use concat_arrays::concat_arrays;

fn main() {
    let arr: [u8; 8] = concat_arrays!([97, 112, 112, 108, 101], [0; 3]);
    println!("{:?}", arr);
}

我认为没有外部依赖是不可能的。

Answer 2

您可以从零开始，然后设置初始值。不过，这需要 arr 为 mut。

fn main() {
    let mut arr: [u8;8] = [0;8];
    let init = [97, 112, 112, 108, 101];
    arr[..init.len()].copy_from_slice(&init);

    println!("Len is {}",arr.len());
    println!("Elements are {:?}",arr);
}

Link 至 Playground

Answer 3

除了其他的答案，你可以使用const泛型来写一个专用的方法。

fn pad_zeroes<const A: usize, const B: usize>(arr: [u8; A]) -> [u8; B] {
    assert!(B >= A); //just for a nicer error message, adding #[track_caller] to the function may also be desirable
    let mut b = [0; B];
    b[..A].copy_from_slice(&arr);
    b
}

Playground

如何用零填充数组？

How to pad an array with zeros?

rust

rust-macros