idiomatic rust

codes/rust/chapter_backtracking/n_queens.rs

@@ -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]; // 记录次对角线上是否有皇后

codes/rust/chapter_backtracking/subset_sum_i.rs

@@ -6,15 +6,15 @@
 
 /* 回溯算法：子集和 I */
 fn backtrack(
- mut state: Vec<i32>,
+ state: &mut Vec<i32>,
  target: i32,
  choices: &[i32],
  start: usize,
  res: &mut Vec<Vec<i32>>,
 ) {
  // 子集和等于 target 时，记录解
  if target == 0 {
- res.push(state);
+ res.push(state.clone());
  return;
  }
  // 遍历所有选择
@@ -28,19 +28,19 @@ 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();
  }
 }
 
 /* 求解子集和 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
 }
 

codes/rust/chapter_backtracking/subset_sum_i_naive.rs

@@ -6,15 +6,15 @@
 
 /* 回溯算法：子集和 I */
 fn backtrack(
- mut state: Vec<i32>,
+ state: &mut Vec<i32>,
  target: i32,
  total: i32,
  choices: &[i32],
  res: &mut Vec<Vec<i32>>,
 ) {
  // 子集和等于 target 时，记录解
  if total == target {
- res.push(state);
+ res.push(state.clone());
  return;
  }
  // 遍历所有选择
@@ -26,18 +26,18 @@ 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();
  }
 }
 
 /* 求解子集和 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
 }
 

codes/rust/chapter_backtracking/subset_sum_ii.rs

@@ -6,15 +6,15 @@
 
 /* 回溯算法：子集和 II */
 fn backtrack(
- mut state: Vec<i32>,
+ state: &mut Vec<i32>,
  target: i32,
  choices: &[i32],
  start: usize,
  res: &mut Vec<Vec<i32>>,
 ) {
  // 子集和等于 target 时，记录解
  if target == 0 {
- res.push(state);
+ res.push(state.clone());
  return;
  }
  // 遍历所有选择
@@ -33,19 +33,19 @@ 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();
  }
 }
 
 /* 求解子集和 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
 }