目标
给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用 一次 。
注意:解集不能包含重复的组合。
示例 1:
输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]
示例 2:
输入: candidates = [2,5,2,1,2], target = 5,
输出:
[
[1,2,2],
[5]
]
说明:
- 1 <= candidates.length <= 100
- 1 <= candidates[i] <= 50
- 1 <= target <= 30
思路
有一个数组 candidates,用 candidates 中的元素值组成目标值 target,返回每种组合的元素列表。数组中的每个元素在每种组合中只能使用一次。
考虑是否选择某个元素,可能的组合有 2^n
种。但是题目有条件限制,要求组合元素值之和等于target,那么如果组合的和大于 target
可以提前返回。因此可以将 candidates 从小到大排序,回溯。关键是如何去除重复组合,对于相同的元素,只关心取的个数,因此如果不选某个元素,后续相同的元素也都不能选。
代码
/**
* @date 2025-01-26 15:24
*/
public class CombinationSum40 {
public List<List<Integer>> combinationSum2(int[] candidates, int target) {
Arrays.sort(candidates);
List<List<Integer>> res = new ArrayList<>();
List<Integer> path = new ArrayList<>();
dfs(0, 0, target, candidates, path, res);
return res;
}
public void dfs(int index, int sum, int target, int[] candidates, List<Integer> path, List<List<Integer>> res) {
if (sum == target) {
List<Integer> tmp = new ArrayList<>(path);
res.add(tmp);
return;
}
if (index == candidates.length || sum > target) {
return;
}
int cur = candidates[index];
path.add(cur);
dfs(index + 1, sum + cur, target, candidates, path, res);
path.remove(path.size() - 1);
index++;
// 如果不选当前的值,也同样不选后续所有相同的值
// 对于相同的值,只考虑选了几个,如果这里不跳过,就会出现重复
// 比如有4个相同的值,选 选 不选 不选,如果允许后续再次选择,就会出现
// 选 不选 不选 选、
// 选 不选 选 不选
// 不选 选 选 不选、
// 不选 选 不选 选、
// 不选 不选 选 选、
while (index < candidates.length && cur == candidates[index]) {
index++;
}
dfs(index, sum, target, candidates, path, res);
}
}