回溯法专题/组合总和Ⅰ/Ⅱ/Ⅲ
39. 组合总和
给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的数字可以无限制重复被选取。
说明:
所有数字(包括 target)都是正整数。
解集不能包含重复的组合。
示例 1:
输入:candidates = [2,3,6,7], target = 7,
所求解集为:
[
[7],
[2,2,3]
]
示例 2:
输入:candidates = [2,3,5], target = 8,
所求解集为:
[
[2,2,2,2],
[2,3,3],
[3,5]
]
提示:
1 <= candidates.length <= 30
1 <= candidates[i] <= 200
candidate 中的每个元素都是独一无二的。
1 <= target <= 500
思路:
自顶向下搜索。
关键点1:candidates 中的数字可以无限制重复被选取。因此,我们在递归到下一层的时候任然要从最初一层的起点开始循环。
关键点2:数组无重复元素,因此在同一层的循环中,当前循环 i i i 结束,我们下一次循环要从下一个数字 i + 1 i+1 i+1 开始,可以保证最后不会出现重复的组合。
代码:
class Solution {
public:
vector<vector<int>> res;
void dfs(vector<int>& candidates, int target,int ind, vector<int> tmp )
{
if(target<0)
return ;
if(target==0)
{
res.emplace_back(tmp);
return ;
}
for(int i=ind;i< candidates.size();i++) //从下一个元素开始,避免组合tmp的重复
{
tmp.emplace_back(candidates[i]);
dfs(candidates,target-candidates[i],i,tmp); //继续从当前元素i开始,因为元素可以重复选取
tmp.pop_back();
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
vector<int> tmp;
dfs(candidates,target,0,tmp);
return res;
}
};
结果:
40. 组合总和 II
给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用一次。
说明:
所有数字(包括目标数)都是正整数。
解集不能包含重复的组合。
示例 1:
输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:
输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]
思路:
与上一题不同的是:
关键点1:candidates 中的数字在一个组合中只能使用一次。因此,我们在递归到下一层的时候要从当前层起点的下一个开始循环。
关键点2:数组可能有重复元素,因此一开始我们要对数组进行排序。然后,在同一层的循环中,对于起点之后的数,要判断其是否与前一个数字一样,若一样,则不进行处理,以此避免出现重复的组合。(同样的处理可以参考链接1)
代码:
class Solution {
public:
vector<vector<int>> res;
void dfs(vector<int>& candidates, int target, int ind, vector<int> tmp)
{
if(target<0)
return ;
if(target==0)
{
res.emplace_back(tmp);
return ;
}
//经过排序的,若减法后有小于0的,则后面的也不用考虑,此处是>=0而不是>0,>0会导致结果空集
//(1)i往下走,避免组合内的重复
for(int i=ind;i<candidates.size() && target-candidates[i]>=0 ;i++ )
{
//(2)i>ind 非常关键,即从起点之后才考虑重复,避免组合内的重复
//(1)+(2)联合起来,才能真正去除组合内的重复
if( i>ind && candidates[i]==candidates[i-1])
continue;
tmp.emplace_back(candidates[i]);
dfs(candidates,target-candidates[i],i+1,tmp); //从下一个元素开始,避免元素的重复选取
tmp.pop_back();
}
}
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(), candidates.end());
vector<int> tmp;
dfs(candidates,target,0,tmp);
return res;
}
};
结果:
216. 组合总和 III
找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。
说明:
所有数字都是正整数。
解集不能包含重复的组合。
示例 1:
输入: k = 3, n = 7
输出: [[1,2,4]]
示例 2:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
思路:
和上一题思路很相似,因为数组元素是1-9,因此无需去重。
代码:
class Solution {
public:
vector<vector<int>> res;
void dfs(int k, int n, int ind, vector<int> tmp )
{
if(k==0 && n!=0)
return ;
if(k==0 && n==0 )
{
res.emplace_back(tmp);
return ;
}
//从下一个元素开始,避免组合tmp的重复,因为数组是有序的,我们可以要求n-i>=0
for(int i=ind;i<=9 && n-i>=0;i++)
{
tmp.emplace_back(i);
dfs(k-1,n-i,i+1,tmp); //从下一个元素开始,避免元素的重复选取
tmp.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
vector<int> tmp;
if(k==0)
return res;
dfs(k,n,1,tmp);
return res;
}
};
结果:
参考链接:
[1] 消失男孩:[LeetCode][C++]三数之和等相关题目