|
| 1 | +// ## The Data Structures ## |
| 2 | + |
| 3 | +// A node in the binary tree. |
| 4 | +struct Node { |
| 5 | + value: i32, |
| 6 | + // Option<Box<Node>> allows for nullable, heap-allocated children. |
| 7 | + left: Option<Box<Node>>, |
| 8 | + right: Option<Box<Node>>, |
| 9 | +} |
| 10 | + |
| 11 | +// Represents the tree itself. |
| 12 | +struct BinaryTree { |
| 13 | + root: Option<Box<Node>>, |
| 14 | +} |
| 15 | + |
| 16 | +// ## Implementation ## |
| 17 | + |
| 18 | +/// A recursive helper function for the post-order traversal. |
| 19 | +fn post_order_recursive(node: &Option<Box<Node>>, result: &mut Vec<i32>) { |
| 20 | + // Check if the current node exists. |
| 21 | + if let Some(current_node) = node { |
| 22 | + // 1. Traverse the left subtree. |
| 23 | + post_order_recursive(¤t_node.left, result); |
| 24 | + |
| 25 | + // 2. Traverse the right subtree. |
| 26 | + post_order_recursive(¤t_node.right, result); |
| 27 | + |
| 28 | + // 3. Visit the root node last. |
| 29 | + result.push(current_node.value); |
| 30 | + } |
| 31 | +} |
| 32 | + |
| 33 | +/// The public function to start the post-order traversal. |
| 34 | +pub fn post_order_traversal(tree: &BinaryTree) -> Vec<i32> { |
| 35 | + let mut result = Vec::new(); |
| 36 | + post_order_recursive(&tree.root, &mut result); |
| 37 | + result |
| 38 | +} |
| 39 | + |
| 40 | +// ## Putting It All Together ## |
| 41 | + |
| 42 | +fn main() { |
| 43 | + // Manually construct a sample binary tree. |
| 44 | + // 4 |
| 45 | + // / \ |
| 46 | + // 2 5 |
| 47 | + // / \ |
| 48 | + // 1 3 |
| 49 | + let tree = BinaryTree { |
| 50 | + root: Some(Box::new(Node { |
| 51 | + value: 4, |
| 52 | + left: Some(Box::new(Node { |
| 53 | + value: 2, |
| 54 | + left: Some(Box::new(Node { |
| 55 | + value: 1, |
| 56 | + left: None, |
| 57 | + right: None, |
| 58 | + })), |
| 59 | + right: Some(Box::new(Node { |
| 60 | + value: 3, |
| 61 | + left: None, |
| 62 | + right: None, |
| 63 | + })), |
| 64 | + })), |
| 65 | + right: Some(Box::new(Node { |
| 66 | + value: 5, |
| 67 | + left: None, |
| 68 | + right: None, |
| 69 | + })), |
| 70 | + })), |
| 71 | + }; |
| 72 | + |
| 73 | + let traversal_result = post_order_traversal(&tree); |
| 74 | + |
| 75 | + println!("Post-order traversal result: {:?}", traversal_result); |
| 76 | +} |
0 commit comments