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
@@ -422,9 +422,32 @@ comments: true
=== "Ruby"
```ruby title="binary_search_recur.rb"
[class]{}-[func]{dfs}
### 二分查找:问题 f(i, j) ###
def dfs(nums, target, i, j)
# 若区间为空,代表无目标元素,则返回 -1
return -1 if i > j
# 计算中点索引 m
m = (i + j) / 2
[class]{}-[func]{binary_search}
if nums[m] < target
# 递归子问题 f(m+1, j)
return dfs(nums, target, m + 1, j)
elsif nums[m] > target
# 递归子问题 f(i, m-1)
return dfs(nums, target, i, m - 1)
else
# 找到目标元素,返回其索引
return m
end
end
### 二分查找 ###
def binary_search(nums, target)
n = nums.length
# 求解问题 f(0, n-1)
dfs(nums, target, 0, n - 1)
end
```
=== "Zig"
@@ -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"
@@ -409,7 +409,7 @@ comments: true
/* 移动一个圆盘 */
fn move_pan(src: &mut Vec<i32>, tar: &mut Vec<i32>) {
// 从 src 顶部拿出一个圆盘
let pan = src.remove(src.len() - 1);
let pan = src.pop().unwrap();
// 将圆盘放入 tar 顶部
tar.push(pan);
}
@@ -510,11 +510,36 @@ comments: true
=== "Ruby"
```ruby title="hanota.rb"
[class]{}-[func]{move}
### 移动一个圆盘 ###
def move(src, tar)
# 从 src 顶部拿出一个圆盘
pan = src.pop
# 将圆盘放入 tar 顶部
tar << pan
end
[class]{}-[func]{dfs}
### 求解汉诺塔问题 f(i) ###
def dfs(i, src, buf, tar)
# 若 src 只剩下一个圆盘,则直接将其移到 tar
if i == 1
move(src, tar)
return
end
[class]{}-[func]{solve_hanota}
# 子问题 f(i-1) :将 src 顶部 i-1 个圆盘借助 tar 移到 buf
dfs(i - 1, src, tar, buf)
# 子问题 f(1) :将 src 剩余一个圆盘移到 tar
move(src, tar)
# 子问题 f(i-1) :将 buf 顶部 i-1 个圆盘借助 src 移到 tar
dfs(i - 1, buf, src, tar)
end
### 求解汉诺塔问题 ###
def solve_hanota(_A, _B, _C)
n = _A.length
# 将 A 顶部 n 个圆盘借助 B 移到 C
dfs(n, _A, _B, _C)
end
```
=== "Zig"