如何在Rust中将树形结构扁平化为Vec<&mut Node>？

你的代码报错核心原因是：执行nodes.push(self)后，后续的if let Node::Dir(dir) = self尝试移动&mut Node引用，违反了Rust可变引用的独占性规则。只需调整模式匹配的方式，就能解决问题，无需引入Rc或RefCell。

修正后的Node扁平化方法 #

将模式匹配改为使用ref mut获取内部Dir的可变引用，而非消耗self引用：

impl Node {
    fn flatten(&mut self, nodes: &mut Vec<&mut Node>) {
        nodes.push(self);

        // 使用 ref mut 借用Dir的可变引用，不移动self
        if let Node::Dir(ref mut dir) = self {
            for node in dir.children.iter_mut() {
                node.flatten(nodes);
            }
        }
    }
}

处理FileSystem的扁平化 #

由于你的FileSystem根节点是Dir而非Node，有两种处理方式：

方案一：调整FileSystem结构（推荐） #

将根节点改为Node，让树形结构更统一：

#[derive(Debug, Clone)]
struct FileSystem {
    root: Node, // 替换原有的Dir类型
}

impl FileSystem {
    fn flatten(&mut self) -> Vec<&mut Node> {
        let mut nodes = Vec::new();
        self.root.flatten(&mut nodes);
        nodes
    }
}

方案二：不修改原有结构（unsafe方式） #

通过指针转换将Dir转为Node引用（仅当你不想调整结构时使用）：

impl FileSystem {
    fn flatten(&mut self) -> Vec<&mut Node> {
        let mut nodes = Vec::new();
        // 安全前提：Node::Dir变体的内存布局与Dir一致，当前场景下合法
        let root_node: &mut Node = unsafe { &mut *( &mut self.root as *mut Dir as *mut Node ) };
        root_node.flatten(&mut nodes);
        nodes
    }
}

补充说明 #

Rc和RefCell在此场景下是多余的：Rc用于多所有权共享，RefCell用于绕过编译期借用检查实现内部可变性，而当前仅需在单线程下遵循正常的可变引用规则即可完成遍历。

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