From 8a6ce26f6a6bdc7897b366bc70ada4bcb4c5e235 Mon Sep 17 00:00:00 2001 From: rongyi Date: Fri, 23 Aug 2024 02:33:47 +0800 Subject: [PATCH] idiomatic rust (#1485) * idomatic rust * More idiomatic rust * make rust code more idiomatic * update --- codes/rust/chapter_backtracking/n_queens.rs | 19 ++++--------------- .../rust/chapter_backtracking/subset_sum_i.rs | 10 +++++----- .../subset_sum_i_naive.rs | 10 +++++----- .../chapter_backtracking/subset_sum_ii.rs | 10 +++++----- 4 files changed, 19 insertions(+), 30 deletions(-) diff --git a/codes/rust/chapter_backtracking/n_queens.rs b/codes/rust/chapter_backtracking/n_queens.rs index 6676f2d12..cf727a13e 100644 --- a/codes/rust/chapter_backtracking/n_queens.rs +++ b/codes/rust/chapter_backtracking/n_queens.rs @@ -16,11 +16,7 @@ fn backtrack( ) { // 当放置完所有行时，记录解 if row == n { - let mut copy_state: Vec> = 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>> { // 初始化 n*n 大小的棋盘，其中 'Q' 代表皇后，'#' 代表空位 - let mut state: Vec> = Vec::new(); - for _ in 0..n { - let mut row: Vec = Vec::new(); - for _ in 0..n { - row.push("#".into()); - } - state.push(row); - } + let mut state: Vec> = 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]; // 记录次对角线上是否有皇后 diff --git a/codes/rust/chapter_backtracking/subset_sum_i.rs b/codes/rust/chapter_backtracking/subset_sum_i.rs index e2a5d0198..052181d60 100644 --- a/codes/rust/chapter_backtracking/subset_sum_i.rs +++ b/codes/rust/chapter_backtracking/subset_sum_i.rs @@ -6,7 +6,7 @@ /* 回溯算法：子集和 I */ fn backtrack( - mut state: Vec, + state: &mut Vec, 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> { - 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 } diff --git a/codes/rust/chapter_backtracking/subset_sum_i_naive.rs b/codes/rust/chapter_backtracking/subset_sum_i_naive.rs index b2dae54ea..ab28a2f6a 100644 --- a/codes/rust/chapter_backtracking/subset_sum_i_naive.rs +++ b/codes/rust/chapter_backtracking/subset_sum_i_naive.rs @@ -6,7 +6,7 @@ /* 回溯算法：子集和 I */ fn backtrack( - mut state: Vec, + state: &mut Vec, 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> { - 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 } diff --git a/codes/rust/chapter_backtracking/subset_sum_ii.rs b/codes/rust/chapter_backtracking/subset_sum_ii.rs index cd7677db5..d3c6f4319 100644 --- a/codes/rust/chapter_backtracking/subset_sum_ii.rs +++ b/codes/rust/chapter_backtracking/subset_sum_ii.rs @@ -6,7 +6,7 @@ /* 回溯算法：子集和 II */ fn backtrack( - mut state: Vec, + state: &mut Vec, 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> { - 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 }