如何在Rust中提取字符串向量的元素?

时间:2019-03-08 03:18:09

标签: vector rust ownership

假设我有以下code

fn extract() -> Vec<String> {
    let data = vec!["aaa".to_string(), "bbb".to_string(), "ccc".to_string()];
    vec![data[0], data[2]]
}

实际上,我从文件中读取了data

很显然,这无法编译,因为我正在从向量data中拉出字符串,从而使向量处于未定义状态。但是,从概念上讲,它应该起作用,因为无论如何我以后都不会使用data

我可以使用mem::replace,但这似乎很疯狂:

fn extract() -> Vec<String> {
    let mut data = vec!["aaa".to_string(), "bbb".to_string(), "ccc".to_string()];
    let a = mem::replace(&mut data[0], "".to_string());
    let c = mem::replace(&mut data[2], "".to_string());
    vec![a, c]
}

我该如何在不克隆字符串的情况下从向量中提取特定元素?

2 个答案:

答案 0 :(得分:4)

Vec为此具有特殊的方法。 swap_removeremove(警告,线性复杂度),drain。例如,

fn extract() -> Vec<String> {
    let mut data = vec!["aaa".to_string(), "bbb".to_string(), "ccc".to_string()];
    // order does matter
    vec![data.swap_remove(2), data.swap_remove(0)]
}

答案 1 :(得分:1)

向量中不能有“孔”。因此,当您从其中移出某些内容时,可以更改其余元素的索引(使用removeswap_remove),或将其替换为其他内容。

为什么不按顺序使用向量,而忽略不需要的向量呢?如果需要保存一些元素以备后用,则可以使用data.iter().filter(...).collect()。如果您真的想避免复制任何字符串,可以将它们包装在Rc中,以便仅复制指针。