Skip to content

Commit

Permalink
More idiomatic rust
Browse files Browse the repository at this point in the history
  • Loading branch information
rongyi committed Aug 16, 2024
1 parent 5b2bd9b commit 42e85e8
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions codes/rust/chapter_backtracking/subset_sum_i_naive.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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;
}
// 遍历所有选择
Expand All @@ -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
}

Expand Down

0 comments on commit 42e85e8

Please sign in to comment.