如何在Rust中正确实现C++ adjacent_difference迭代器算法？

你要实现的是类似C++的adjacent_difference算法，输入[5, 7, 16]应输出[5, 2, 9]。原实现里用i - <I as Iterator>::Item::default()生成第一个元素的写法确实不够优雅，还强行依赖了Default trait，限制了迭代器的适用范围（比如无法处理没有合理Default实现的自定义类型）。

优化思路 #

问题核心是第一个元素需要从原迭代器的Item类型转换为Sub::Output类型，而非通过减法强行构造。我们可以通过添加Into<<I::Item as Sub>::Output>约束，让原Item能安全转换为差值类型，替代不合理的Default依赖。

优化后的代码 #

struct AdjacentDifference<I>
where I: Iterator
{
    prev: Option<I::Item>,
    it: I,
}

impl<I> Iterator for AdjacentDifference<I>
where 
    I: Iterator,
    I::Item: std::ops::Sub + Copy + Into<<I::Item as std::ops::Sub>::Output>,
{
    type Item = <<I::Item as std::ops::Sub>::Output>;

    fn next(&mut self) -> Option<Self::Item> {
        match (self.it.next(), self.prev.take()) {
            (None, _) => None,
            (Some(current), Some(prev_val)) => {
                self.prev = Some(current);
                Some(current - prev_val)
            }
            (Some(first), None) => {
                self.prev = Some(first);
                Some(first.into())
            }
        }
    }
}

trait AdjacentDifferenceExt: Iterator where Self: Sized {
    fn adjacent_difference(self) -> AdjacentDifference<Self>
    where 
        Self::Item: std::ops::Sub + Copy + Into<<Self::Item as std::ops::Sub>::Output>,
    {
        AdjacentDifference {
            prev: None,
            it: self,
        }
    }
}

impl<I> AdjacentDifferenceExt for I where I: Iterator {}

关键改进点 #

1. 移除Default约束：不再依赖Item的默认值构造第一个元素，避免了语义不匹配的问题（比如自定义类型的Default可能不是减法单位元）。
2. 使用类型转换：通过Into trait将第一个元素直接转换为差值类型，完全符合C++算法的原始语义（第一个元素保留输入值）。
3. 简化状态处理：用prev.take()替代直接匹配self.prev，逻辑更清晰直观。

额外优化（可选） #

如果希望支持更灵活的差值计算（比如自定义二元操作，对应C++的重载版本），可以给迭代器添加一个闭包参数，让用户自定义相邻元素的计算逻辑：

struct AdjacentDifference<I, F>
where
    I: Iterator,
    F: FnMut(&I::Item, &I::Item) -> <I::Item as std::ops::Sub>::Output,
{
    prev: Option<I::Item>,
    it: I,
    op: F,
}

// 对应的Iterator实现和trait扩展可参考上述逻辑修改

内容的提问来源于stack exchange，提问作者skelix