vector - 如何连接两个切片或两个向量并仍然可以访问原始值？

Question

我有两个切片或向量，我想添加它们，如 Golang 中所示:

a := []byte{1, 2, 3}
b := []byte{4, 5, 6}
ab := append(a, b...)
ba := append(b, a...)

我如何在 Rust 中做到这一点？我发现了一些其他问题，例如:

• What's the idiomatic way to append a slice to a vector?
• Best way to concatenate vectors in Rust

但是，他们所有的最佳答案都是a += b，而不是ab = a + b。

let mut a = vec![1, 2, 3];
let mut b = vec![4, 5, 6];

a.append(&mut b);

assert_eq!(a, [1, 2, 3, 4, 5, 6]);
assert_eq!(b, []);

或者 Rust 中是否有类似 Vec::append(a, b) 的函数？

Answer 1

您可以 chain你的迭代器:

fn main() {
    let a = vec![1, 2, 3];
    let b = vec![4, 5, 6];

    // Don't consume the original vectors and clone the items:
    let ab: Vec<_> = a.iter().chain(&b).cloned().collect();

    // Consume the original vectors. The items do not need to be cloneable:
    let ba: Vec<_> = b.into_iter().chain(a).collect();

    assert_eq!(ab, [1, 2, 3, 4, 5, 6]);
    assert_eq!(ba, [4, 5, 6, 1, 2, 3]);
}

请注意，迭代器知道它产生的项数，因此 collect 可以直接分配正确数量的内存:

fn main() {
    let a = vec![1, 2, 3];
    let b = vec![4, 5, 6];

    let ba = b.into_iter().chain(a);
    assert_eq!(ba.size_hint(), (6, Some(6)));

    let ba: Vec<_> = ba.collect();
    assert_eq!(ba, [4, 5, 6, 1, 2, 3]);
}