Как я могу объединить два среза или два вектора и при этом иметь доступ к исходным значениям? - PullRequest
Новая
       45

Как я могу объединить два среза или два вектора и при этом иметь доступ к исходным значениям?

0 голосов
/ 11 января 2019

У меня есть два среза или вектора, и я хочу добавить их, как показано здесь, на Голанге: 

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)?

Ответы [ 2 ]

0 голосов
/ 11 января 2019

Вы можете 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]);
}
0 голосов
/ 11 января 2019

Нет конструктора 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);.

...