This commit is contained in:
krahets
2024-05-15 19:00:27 +08:00
parent bd54cd096b
commit e434a3343c
36 changed files with 402 additions and 107 deletions
@@ -485,9 +485,31 @@ comments: true
=== "Ruby"
```ruby title="build_tree.rb"
[class]{}-[func]{dfs}
### 构建二叉树:分治 ###
def dfs(preorder, inorder_map, i, l, r)
# 子树区间为空时终止
return if r - l < 0
[class]{}-[func]{build_tree}
# 初始化根节点
root = TreeNode.new(preorder[i])
# 查询 m ,从而划分左右子树
m = inorder_map[preorder[i]]
# 子问题:构建左子树
root.left = dfs(preorder, inorder_map, i + 1, l, m - 1)
# 子问题:构建右子树
root.right = dfs(preorder, inorder_map, i + 1 + m - l, m + 1, r)
# 返回根节点
root
end
### 构建二叉树 ###
def build_tree(preorder, inorder)
# 初始化哈希表,存储 inorder 元素到索引的映射
inorder_map = {}
inorder.each_with_index { |val, i| inorder_map[val] = i }
dfs(preorder, inorder_map, 0, 0, inorder.length - 1)
end
```
=== "Zig"