给定一个候选人编号的集合 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、解集不能包含重复的组合。
2、candidates
中的每个数字在每个组合中只能使用 一次 。
在遍历时要对这两种情况进行处理。
进入回溯三部曲(递归函数参数,递归终止条件,单层搜索逻辑)
1、递归函数参数
vector<vector<int>> res;
vector<int> path;
void backtracking(vector<int>& candidates, int target, int sum, int start)
其中res是我们返回的结果集,path是我们用来存储搜索到的所有可能的结果、sum是每次累加之后的结果,start表示从哪里开始取元素。
2、递归终止条件
显然当累加和sum等于target时终止循环,并将此时的结果集存入res
if (sum == target) {
res.push_back(path);
return;
}
3、单层搜索逻辑
首先进行一下剪枝操作,当sum+当前的candidates[i]大于target时(已排序),表示后面也不会再有合适的结果集了,可以直接跳出。
再进行去重操作,即相等的元素只使用一次,只需判断前后元素是否相等,相等则跳出此次循环
下面就是回溯的过程了:
sum += candidates[i];
path.push_back(candidates[i]);
backtracking(candidates, target, sum, i + 1);
sum -= candidates[i];
path.pop_back();
假如数据为
[10,1,2,7,6,1,5]
和8
排序后的数组为:
[1,1,2,5,6,7,10]
假设此时path内元素为1,1;此时将2push后将5push发现比target大,返回到push5的循环中,此后6,7,10都将跳出,然后回到push2的循环中,将2给pop,push5发现和目标值相等,添加并返回,此后6,7,10都将不满足跳出。然后回到第二个1的循环中,然后将2,5,6,7,10都和第一个1组合一次,回溯完成。
for (int i = start; i < candidates.size(); i++) {
if (sum + candidates[i] > target) {
return;
}
if (i > start && candidates[i] == candidates[i-1]) {
continue; // 跳过重复的候选数
}
sum += candidates[i];
path.push_back(candidates[i]);
backtracking(candidates, target, sum, i + 1);
sum -= candidates[i];
path.pop_back();
}
整体C++代码如下
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
void backtracking(vector<int>& candidates, int target, int sum, int start) {
if (sum == target) {
res.push_back(path);
return;
}
for (int i = start; i < candidates.size(); i++) {
if (sum + candidates[i] > target) {
return;
}
if (i > start && candidates[i] == candidates[i-1]) {
continue; // 跳过重复的候选数
}
sum += candidates[i];
path.push_back(candidates[i]);
backtracking(candidates, target, sum, i + 1);
sum -= candidates[i];
path.pop_back();
}
}
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(), candidates.end()); // 对候选数进行排序
backtracking(candidates, target, 0, 0);
return res;
}
};