如何从HashSet形成切片?

时间:2018-05-08 14:34:29

标签: rust lifetime

结构定义为:

struct Node {
    set: HashSet<usize>,
    // other fields omitted
}

必须为特征实现一个函数(兼容性问题),需要将集合中的所有元素作为切片返回。

我知道以下功能不起作用:

impl Node {
    pub fn set_slice(&self) -> &[usize] {
        let elems: Vec<_> = self.set.iter().cloned().collect();
        &elems[..]
    }
}

问题是:

error[E0597]: `elems` does not live long enough
  --> src/main.rs:11:10
   |
11 |         &elems[..]
   |          ^^^^^ borrowed value does not live long enough
12 |     }
   |     - borrowed value only lives until here
   |
note: borrowed value must be valid for the anonymous lifetime #1 defined on the method body at 9:5...
  --> src/main.rs:9:5
   |
9  | /     pub fn set_slice(&self) -> &[usize] {
10 | |         let elems: Vec<_> = self.set.iter().cloned().collect();
11 | |         &elems[..]
12 | |     }
   | |_____^

我知道这个要求可能听起来很奇怪。尽管为什么我必须这样做,有没有“好”的方法来实现这个目标?

如果可能的话,我想保留HashSet容器进行O(1)查找,我不想引入新的struct成员来节省内存。

2 个答案:

答案 0 :(得分:5)

不,在安全的Rust中你的要求是完全不可能的。

HashSet / HashMap没有连续的数据集合,因此无法从中获取切片。

如果你可以改变一些事情,那么你就有了选择。

如果您可以存储HashSet并且方法为Vec,则可以“呈现&mut self的视图”:

struct Node {
    set: HashSet<usize>,
    view: Vec<usize>,
    // other fields omitted
}

impl Node {
    pub fn set_slice(&mut self) -> &[usize] {
        self.view.clear();
        self.view.extend(self.set.iter().cloned());
        &self.view
    }
}

您可以返回借用或拥有的Cow

use std::borrow::Cow;

impl Node {
    pub fn set_slice(&self) -> Cow<[usize]> {
        self.set.iter().cloned().collect::<Vec<_>>().into()
    }
}

你可以return an iterator over the values

impl Node {
    pub fn set_slice<'a>(&'a self) -> impl Iterator<Item = &'a usize> + 'a {
        self.set.iter()
    }
}

可能 a crate使用紧凑的Vec作为其后备存储,然后可以将其作为切片公开。

答案 1 :(得分:0)

这在简单(基本)方面是不可能的。

这可以通过Boxmut static实现,但我建议您修改自己的特性并返回类似以下示例的内容:

您可以在特质中使用AsRef<[T]>代替&[usize]。或者只是返回一个迭代器。

struct Node {
    set: HashSet<usize>,
}

trait SetSlice {
    type Slice: AsRef<[usize]>;
    fn get_slice_cloned(&self) -> Self::Slice;
}

impl SetSlice for Node {
    type Slice = Vec<usize>;
    fn get_slice_cloned(&self) -> Self::Slice { self.set.iter().cloned().collect() }
}

// there we use auto-impl of Iterator trait
// and return the iter.
// NOTE: we cannot use auto-impl in trait methods.
impl Node {
    fn get_neat_iter(&self) -> impl Iterator<Item = &usize> { self.set.iter() }
}

fn need_slice(slice: &[usize]) {}

fn main() {
    let n = Node { set: Default::default(), };

    // as_ref
    let all = n.get_slice_cloned();
    need_slice(all.as_ref());

    // iter-way
    let all: Vec<_> = n.get_neat_iter().cloned().collect();
    need_slice(&all);
}

这只有两种方式。