如何将＆HashSet <＆T>用作IntoIterator <Item =＆T>？

Question

我有一个函数，它需要一个&T（由IntoIterator表示）的集合，要求每个元素都是唯一的。

fn foo<'a, 'b, T: std::fmt::Debug, I>(elements: &'b I)
where
    &'b I: IntoIterator<Item = &'a T>,
    T: 'a,
    'b: 'a,

我还想编写一个包装函数，即使元素不是唯一的，也可以通过使用HashSet首先删除重复的元素来工作。

我尝试了以下实现：

use std::collections::HashSet;

fn wrap<'a, 'b, T: std::fmt::Debug + Eq + std::hash::Hash, J>(elements: &'b J)
where
    &'b J: IntoIterator<Item = &'a T>,
    T: 'a,
    'b: 'a,
{
    let hashset: HashSet<&T> = elements.into_iter().into_iter().collect();
    foo(&hashset);
}

playground

但是，编译器对我的假设HashSet<&T>实现IntoIterator<Item = &'a T>并不满意：

error[E0308]: mismatched types
  --> src/lib.rs:10:9
   |
10 |     foo(&hashset);
   |         ^^^^^^^^ expected type parameter, found struct `std::collections::HashSet`
   |
   = note: expected type `&J`
              found type `&std::collections::HashSet<&T>`
   = help: type parameters must be constrained to match other types
   = note: for more information, visit https://doc.rust-lang.org/book/ch10-02-traits.html#traits-as-parameters

我知道我可以通过克隆所有输入元素来使用HashSet<T>，但是我想避免不必要的复制和内存使用。

Answer 1

如果您有&HashSet<&T>，并且需要可以处理多次的&T（而不是&&T）的迭代器，则可以使用Iterator::copied来转换迭代器的&&T到&T：

use std::{collections::HashSet, fmt::Debug, hash::Hash, marker::PhantomData};

struct Collection<T> {
    item: PhantomData<T>,
}

impl<T> Collection<T>
where
    T: Debug,
{
    fn foo<'a, I>(elements: I) -> Self
    where
        I: IntoIterator<Item = &'a T> + Clone,
        T: 'a,
    {
        for element in elements.clone() {
            println!("{:?}", element);
        }
        for element in elements {
            println!("{:?}", element);
        }
        Self { item: PhantomData }
    }
}

impl<T> Collection<T>
where
    T: Debug + Eq + Hash,
{
    fn wrap<'a, I>(elements: I) -> Self
    where
        I: IntoIterator<Item = &'a T>,
        T: 'a,
    {
        let set: HashSet<_> = elements.into_iter().collect();
        Self::foo(set.iter().copied())
    }
}

#[derive(Debug, Hash, PartialEq, Eq)]
struct Foo(i32);

fn main() {
    let v = vec![Foo(1), Foo(2), Foo(4)];
    Collection::<Foo>::wrap(&v);
}

另请参阅：

• Using the same iterator multiple times in Rust
• Does cloning an iterator copy the entire underlying vector?
• Why does cloning my custom type result in &T instead of T?

---

  

请注意，此答案的其余部分均假设一个名为Collection<T>的结构是T类型的值的集合。 OP已澄清这是不正确的。

这不是您的问题，如后面的示例所示。可以归结为：

struct Collection<T>(T);

impl<T> Collection<T> {
    fn new(value: &T) -> Self {
        Collection(value)
    }
}

您正在引用类型（&T）并尝试将其存储在需要T的位置；这些是不同的类型，将产生错误。您出于某种原因使用PhantomData并通过迭代器接受引用，但是问题是相同的。

实际上，PhantomData使得问题更难发现，因为您可以弥补不起作用的值。例如，我们这里从来没有任何类型的字符串，但是我们“成功”创建了该结构：

use std::marker::PhantomData;

struct Collection<T>(PhantomData<T>);

impl Collection<String> {
    fn new<T>(value: &T) -> Self {
        Collection(PhantomData)
    }
}

最终，您的wrap函数也没有意义：

impl<T: Eq + Hash> Collection<T> {
    fn wrap<I>(elements: I) -> Self
    where
        I: IntoIterator<Item = T>,

这等效于

impl<T: Eq + Hash> Collection<T> {
    fn wrap<I>(elements: I) -> Collection<T>
    where
        I: IntoIterator<Item = T>,

这说，给定元素T的迭代器，您将返回这些元素的集合。但是，您将它们放在HashMap中并对其进行引用迭代，从而产生&T。因此，此功能签名不正确。

您似乎最有可能希望接受拥有值的迭代器：

use std::{collections::HashSet, fmt::Debug, hash::Hash};

struct Collection<T> {
    item: T,
}

impl<T> Collection<T> {
    fn foo<I>(elements: I) -> Self
    where
        I: IntoIterator<Item = T>,
        for<'a> &'a I: IntoIterator<Item = &'a T>,
        T: Debug,
    {
        for element in &elements {
            println!("{:?}", element);
        }
        for element in &elements {
            println!("{:?}", element);
        }

        Self {
            item: elements.into_iter().next().unwrap(),
        }
    }
}

impl<T> Collection<T>
where
    T: Eq + Hash,
{
    fn wrap<I>(elements: I) -> Self
    where
        I: IntoIterator<Item = T>,
        T: Debug,
    {
        let s: HashSet<_> = elements.into_iter().collect();
        Self::foo(s)
    }
}

#[derive(Debug, Hash, PartialEq, Eq)]
struct Foo(i32);

fn main() {
    let v = vec![Foo(1), Foo(2), Foo(4)];
    let c = Collection::wrap(v);
    println!("{:?}", c.item)
}

在这里，我们直接在通用迭代器类型上放置一个特征绑定，并在对迭代器的引用上放置一个第二个高等级特征绑定。这样，我们就可以将对迭代器的引用用作迭代器本身。

另请参阅：

• How does one generically duplicate a value in Rust?
• Is there any way to return a reference to a variable created in a function?
• How do I write the lifetimes for references in a type constraint when one of them is a local reference?

Answer 2

Shepmaster指出，我的代码存在许多正交问题，但是为了解决使用HashSet<&T>作为IntoIterator<Item=&T>的问题，我发现一种解决方法是使用包装器结构：

struct Helper<T, D: Deref<Target = T>>(HashSet<D>);

struct HelperIter<'a, T, D: Deref<Target = T>>(std::collections::hash_set::Iter<'a, D>);

impl<'a, T, D: Deref<Target = T>> Iterator for HelperIter<'a, T, D>
where
    T: 'a,
{
    type Item = &'a T;
    fn next(&mut self) -> Option<Self::Item> {
        self.0.next().map(|x| x.deref())
    }
}

impl<'a, T, D: Deref<Target = T>> IntoIterator for &'a Helper<T, D> {
    type Item = &'a T;
    type IntoIter = HelperIter<'a, T, D>;
    fn into_iter(self) -> Self::IntoIter {
        HelperIter((&self.0).into_iter())
    }
}

用途如下：

struct Collection<T> {
    item: PhantomData<T>,
}

impl<T: Debug> Collection<T> {
    fn foo<I>(elements: I) -> Self
    where
        I: IntoIterator + Copy,
        I::Item: Deref<Target = T>,
    {
        for element in elements {
            println!("{:?}", *element);
        }
        for element in elements {
            println!("{:?}", *element);
        }
        return Self { item: PhantomData };
    }
}

impl<T: Debug + Eq + Hash> Collection<T> {
    fn wrap<I>(elements: I) -> Self
    where
        I: IntoIterator + Copy,
        I::Item: Deref<Target = T> + Eq + Hash,
    {
        let helper = Helper(elements.into_iter().collect());
        Self::foo(&helper);
        return Self { item: PhantomData };
    }
}

fn main() {
    let v = vec![Foo(1), Foo(2), Foo(4)];
    Collection::<Foo>::wrap(&v);
}

我猜想其中有些可能比需要的复杂，但我不确定如何。

full playground