mirror of
https://github.com/krahets/hello-algo.git
synced 2024-12-26 15:26:28 +08:00
50 lines
1.4 KiB
Rust
50 lines
1.4 KiB
Rust
/*
|
|
* File: preorder_traversal_ii_compact.rs
|
|
* Created Time: 2023-07-15
|
|
* Author: sjinzh (sjinzh@gmail.com)
|
|
*/
|
|
|
|
include!("../include/include.rs");
|
|
|
|
use std::{cell::RefCell, rc::Rc};
|
|
use tree_node::{vec_to_tree, TreeNode};
|
|
|
|
/* 前序遍历:例题二 */
|
|
fn pre_order(res: &mut Vec<Vec<Rc<RefCell<TreeNode>>>>, path: &mut Vec<Rc<RefCell<TreeNode>>>, root: Option<Rc<RefCell<TreeNode>>>) {
|
|
if root.is_none() {
|
|
return;
|
|
}
|
|
if let Some(node) = root {
|
|
// 尝试
|
|
path.push(node.clone());
|
|
if node.borrow().val == 7 {
|
|
// 记录解
|
|
res.push(path.clone());
|
|
}
|
|
pre_order(res, path, node.borrow().left.clone());
|
|
pre_order(res, path, node.borrow().right.clone());
|
|
// 回退
|
|
path.remove(path.len() - 1);
|
|
}
|
|
}
|
|
|
|
/* Driver Code */
|
|
pub fn main() {
|
|
let root = vec_to_tree([1, 7, 3, 4, 5, 6, 7].map(|x| Some(x)).to_vec());
|
|
println!("初始化二叉树");
|
|
print_util::print_tree(root.as_ref().unwrap());
|
|
|
|
// 前序遍历
|
|
let mut path = Vec::new();
|
|
let mut res = Vec::new();
|
|
pre_order(&mut res, &mut path, root);
|
|
|
|
println!("\n输出所有根节点到节点 7 的路径");
|
|
for path in res {
|
|
let mut vals = Vec::new();
|
|
for node in path {
|
|
vals.push(node.borrow().val)
|
|
}
|
|
println!("{:?}", vals);
|
|
}
|
|
}
|