hello-algo/codes/rust/chapter_tree/binary_tree.rs
rongyi 7b1094318b
cargo fmt rust code (#1131)
* cargo fmt code

* Add empty line to seperate unrelated comments

* Fix review

* Update bubble_sort.rs

* Update merge_sort.rs

---------

Co-authored-by: Yudong Jin <krahets@163.com>
2024-03-16 02:13:41 +08:00

39 lines
1.1 KiB
Rust

/**
* File: binary_tree.rs
* Created Time: 2023-02-27
* Author: xBLACKICEx (xBLACKICE@outlook.com)
*/
use std::rc::Rc;
include!("../include/include.rs");
use tree_node::TreeNode;
/* Driver Code */
fn main() {
/* 初始化二叉树 */
// 初始化节点
let n1 = TreeNode::new(1);
let n2 = TreeNode::new(2);
let n3 = TreeNode::new(3);
let n4 = TreeNode::new(4);
let n5 = TreeNode::new(5);
// 构建节点之间的引用(指针)
n1.borrow_mut().left = Some(Rc::clone(&n2));
n1.borrow_mut().right = Some(Rc::clone(&n3));
n2.borrow_mut().left = Some(Rc::clone(&n4));
n2.borrow_mut().right = Some(Rc::clone(&n5));
println!("\n初始化二叉树\n");
print_util::print_tree(&n1);
// 插入节点与删除节点
let p = TreeNode::new(0);
// 在 n1 -> n2 中间插入节点 P
p.borrow_mut().left = Some(Rc::clone(&n2));
n1.borrow_mut().left = Some(Rc::clone(&p));
println!("\n插入节点 P 后\n");
print_util::print_tree(&n1);
// 删除节点 P
drop(p);
n1.borrow_mut().left = Some(Rc::clone(&n2));
println!("\n删除节点 P 后\n");
print_util::print_tree(&n1);
}