我有两个切片或向量,我想添加它们,如Golang所示:
a := []byte{1, 2, 3}
b := []byte{4, 5, 6}
ab := append(a, b...)
ba := append(b, a...)
如何在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)
这样的函数?
答案 0 :(得分:6)
您可以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]);
}
答案 1 :(得分:4)
没有Vec的构造函数直接附加两个切片。
该功能不是基本功能,因为您可以自己实现:
let ab = {
let mut r = a.clone();
r.extend_from_slice(&b);
r
};
如果您经常发现自己正在执行此操作,则可能更喜欢编写一个函数来执行此操作:
fn cat<T: Clone>(a: &[T], b: &[T]) -> Vec<T> {
let mut v = Vec::with_capacity(a.len() + b.len());
v.extend_from_slice(a);
v.extend_from_slice(b);
v
}
然后您将能够执行:let ab = cat(&a, &b);
。