build

chapter_divide_and_conquer/build_binary_tree_problem.md

@@ -282,6 +282,37 @@ status: new
     [class]{}-[func]{buildTree}
     ```
 
+=== "Rust"
+
+    ```rust title="build_tree.rs"
+    /* 构建二叉树：分治 */
+    fn dfs(preorder: &[i32], inorder: &[i32], hmap: &HashMap<i32, i32>, i: i32, l: i32, r: i32) -> Option<Rc<RefCell<TreeNode>>> {
+        // 子树区间为空时终止
+        if r - l < 0 { return None; }
+        // 初始化根节点
+        let root = TreeNode::new(preorder[i as usize]);
+        // 查询 m ，从而划分左右子树
+        let m = hmap.get(&preorder[i as usize]).unwrap();
+        // 子问题：构建左子树
+        root.borrow_mut().left = dfs(preorder, inorder, hmap, i + 1, l, m - 1);
+        // 子问题：构建右子树
+        root.borrow_mut().right = dfs(preorder, inorder, hmap, i + 1 + m - l, m + 1, r);
+        // 返回根节点
+        Some(root)
+    }
+
+    /* 构建二叉树 */
+    fn build_tree(preorder: &[i32], inorder: &[i32]) -> Option<Rc<RefCell<TreeNode>>> {
+        // 初始化哈希表，存储 inorder 元素到索引的映射
+        let mut hmap: HashMap<i32, i32> = HashMap::new();
+        for i in 0..inorder.len() {
+            hmap.insert(inorder[i], i as i32);
+        }
+        let root = dfs(preorder, inorder, &hmap, 0, 0, inorder.len() as i32 - 1);
+        root
+    }
+    ```
+
 下图展示了构建二叉树的递归过程，各个节点是在向下“递”的过程中建立的，而各条边（即引用）是在向上“归”的过程中建立的。
 
 === "<1>"