0033-Rust-实现递归迭代

环境

• Time 2022-05-21
• Rust 1.61.0

前言

说明

参考：https://fasterthanli.me/articles/recursive-iterators-rust

目标

对于递归类型的结构，实现递归迭代。

自定义类型

struct Node<T> {
    values: Vec<T>,
    children: Vec<Node<T>>,
}

values

impl<T> Node<T> {
    pub fn values<'a>(&'a self) -> Box<dyn Iterator<Item = &T> + 'a> {
        let chain = self.children.iter().map(|n| n.values()).flatten();
        Box::new(self.values.iter().chain(chain))
    }
}

使用迭代器

fn main() {
    let node = Node {
        values: vec![1, 2, 3],
        children: vec![
            Node {
                values: vec![4, 5, 6],
                children: vec![],
            },
            Node {
                values: vec![7, 8, 9],
                children: vec![],
            },
        ],
    };

    for ele in node.values() {
        println!("{ele:?}")
    }
}

总结

对递归类型的结构体，实现了递归迭代。

附录