xmengnet/hello-algo
1
0
Fork 0
mirror of https://github.com/krahets/hello-algo.git synced 2024-12-24 04:26:30 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

idiomatic rust (#1485)

Browse source 
* idomatic rust

* More idiomatic rust

* make rust code more idiomatic

* update
This commit is contained in:
rongyi 2024-08-23 02:33:47 +08:00 committed by GitHub
parent 6b2c38cae4
commit 8a6ce26f6a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 19 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
codes/rust/chapter_backtracking/n_queens.rs
View file

@ -16,11 +16,7 @@ fn backtrack(
) {
// 当放置完所有行时，记录解
if row == n {
let mut copy_state: Vec<Vec<String>> = Vec::new();
for s_row in state.clone() {
copy_state.push(s_row);
}
res.push(copy_state);
res.push(state.clone());
return;
}
// 遍历所有列
@ -31,12 +27,12 @@ fn backtrack(
// 剪枝：不允许该格子所在列、主对角线、次对角线上存在皇后
if !cols[col] && !diags1[diag1] && !diags2[diag2] {
// 尝试：将皇后放置在该格子
state.get_mut(row).unwrap()[col] = "Q".into();
state[row][col] = "Q".into();
(cols[col], diags1[diag1], diags2[diag2]) = (true, true, true);
// 放置下一行
backtrack(row + 1, n, state, res, cols, diags1, diags2);
// 回退：将该格子恢复为空位
state.get_mut(row).unwrap()[col] = "#".into();
state[row][col] = "#".into();
(cols[col], diags1[diag1], diags2[diag2]) = (false, false, false);
}
}
@ -45,14 +41,7 @@ fn backtrack(
/* 求解 n 皇后 */
fn n_queens(n: usize) -> Vec<Vec<Vec<String>>> {
// 初始化 n*n 大小的棋盘，其中 'Q' 代表皇后，'#' 代表空位
let mut state: Vec<Vec<String>> = Vec::new();
for _ in 0..n {
let mut row: Vec<String> = Vec::new();
for _ in 0..n {
row.push("#".into());
}
state.push(row);
}
let mut state: Vec<Vec<String>> = vec![vec!["#".to_string(); n]; n];
let mut cols = vec![false; n]; // 记录列是否有皇后
let mut diags1 = vec![false; 2 * n - 1]; // 记录主对角线上是否有皇后
let mut diags2 = vec![false; 2 * n - 1]; // 记录次对角线上是否有皇后

10
codes/rust/chapter_backtracking/subset_sum_i.rs
View file

@ -6,7 +6,7 @@
/* 回溯算法：子集和 I */
fn backtrack(
mut state: Vec<i32>,
state: &mut Vec<i32>,
target: i32,
choices: &[i32],
start: usize,
@ -14,7 +14,7 @@ fn backtrack(
) {
// 子集和等于 target 时，记录解
if target == 0 {
res.push(state);
res.push(state.clone());
return;
}
// 遍历所有选择
@ -28,7 +28,7 @@ fn backtrack(
// 尝试：做出选择，更新 target, start
state.push(choices[i]);
// 进行下一轮选择
backtrack(state.clone(), target - choices[i], choices, i, res);
backtrack(state, target - choices[i], choices, i, res);
// 回退：撤销选择，恢复到之前的状态
state.pop();
}
@ -36,11 +36,11 @@ fn backtrack(
/* 求解子集和 I */
fn subset_sum_i(nums: &mut [i32], target: i32) -> Vec<Vec<i32>> {
let state = Vec::new(); // 状态（子集）
let mut state = Vec::new(); // 状态（子集）
nums.sort(); // 对 nums 进行排序
let start = 0; // 遍历起始点
let mut res = Vec::new(); // 结果列表（子集列表）
backtrack(state, target, nums, start, &mut res);
backtrack(&mut state, target, nums, start, &mut res);
res
}

10
codes/rust/chapter_backtracking/subset_sum_i_naive.rs
View file

@ -6,7 +6,7 @@
/* 回溯算法：子集和 I */
fn backtrack(
mut state: Vec<i32>,
state: &mut Vec<i32>,
target: i32,
total: i32,
choices: &[i32],
@ -14,7 +14,7 @@ fn backtrack(
) {
// 子集和等于 target 时，记录解
if total == target {
res.push(state);
res.push(state.clone());
return;
}
// 遍历所有选择
@ -26,7 +26,7 @@ fn backtrack(
// 尝试：做出选择，更新元素和 total
state.push(choices[i]);
// 进行下一轮选择
backtrack(state.clone(), target, total + choices[i], choices, res);
backtrack(state, target, total + choices[i], choices, res);
// 回退：撤销选择，恢复到之前的状态
state.pop();
}
@ -34,10 +34,10 @@ fn backtrack(
/* 求解子集和 I（包含重复子集） */
fn subset_sum_i_naive(nums: &[i32], target: i32) -> Vec<Vec<i32>> {
let state = Vec::new(); // 状态（子集）
let mut state = Vec::new(); // 状态（子集）
let total = 0; // 子集和
let mut res = Vec::new(); // 结果列表（子集列表）
backtrack(state, target, total, nums, &mut res);
backtrack(&mut state, target, total, nums, &mut res);
res
}

10
codes/rust/chapter_backtracking/subset_sum_ii.rs
View file

@ -6,7 +6,7 @@
/* 回溯算法：子集和 II */
fn backtrack(
mut state: Vec<i32>,
state: &mut Vec<i32>,
target: i32,
choices: &[i32],
start: usize,
@ -14,7 +14,7 @@ fn backtrack(
) {
// 子集和等于 target 时，记录解
if target == 0 {
res.push(state);
res.push(state.clone());
return;
}
// 遍历所有选择
@ -33,7 +33,7 @@ fn backtrack(
// 尝试：做出选择，更新 target, start
state.push(choices[i]);
// 进行下一轮选择
backtrack(state.clone(), target - choices[i], choices, i, res);
backtrack(state, target - choices[i], choices, i, res);
// 回退：撤销选择，恢复到之前的状态
state.pop();
}
@ -41,11 +41,11 @@ fn backtrack(
/* 求解子集和 II */
fn subset_sum_ii(nums: &mut [i32], target: i32) -> Vec<Vec<i32>> {
let state = Vec::new(); // 状态（子集）
let mut state = Vec::new(); // 状态（子集）
nums.sort(); // 对 nums 进行排序
let start = 0; // 遍历起始点
let mut res = Vec::new(); // 结果列表（子集列表）
backtrack(state, target, nums, start, &mut res);
backtrack(&mut state, target, nums, start, &mut res);
res
}