组合总和
题目
给定一个无重复元素的数组 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]
]
思路
一般需要列出所有组合情况的问题都可以用到深度优先遍历来解决,即一条路找到底,然后返回上一层然后继续走到到底,一直到遍历所有情况,过程中将符合情况的记录下来然后返回。
因为这道题是找到组合和与target相等的,而且数字可以重复使用,所以每选择一个数就让target-当前下标对应的值,然后将所得的数传入下去,直到等于0,即找到一个组合,依次递归。
代码
class Solution {
public:
void dfs(vector<vector<int>>& ans,vector<int> &candidates,vector<int> &tmp,int target,int start)//深度优先遍历,start为当前下标
{
if(target<0)//这个组合不对,返回
{
return;
}
else if(target==0)//找到,存在ans中
{
ans.push_back(tmp);
return;
}
else
{
for(int i=start;i<candidates.size();i++)//循环迭代数组元素
{
tmp.push_back(candidates[i]);//将这个值存在临时组合中
dfs(ans,candidates,tmp,target-candidates[i],i);//因为数字何以重复,所以每一次都将当前下标传到下一个递归
tmp.pop_back();//找到或没找到,返回后都将该点弹出以继续遍历下一个
}
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target)
{
vector<vector<int>> ans;//返回值
vector<int> tmp;//临时组合
if(candidates.empty())//为空,反回
{
return ans;
}
sort(candidates.begin(),candidates.end());//排序后组合比较好找
dfs(ans,candidates,tmp,target,0);//dfs
return ans;
}
};
组合总和 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]
]
思路
这一道题与上一道题的区别就是每一个数字不能重复使用,且不能重复,整体思路与上一题类似,只需将相同的值过滤掉便好。
代码
class Solution {
public:
void dfs(vector<vector<int>>& ans,vector<int>& candidates,vector<int>& tmp,int target,int start)
{
if(target==0)
{
ans.push_back(tmp);
return;
}
else if(target < 0)
{
return;
}
else
{
for(int i=start;i<candidates.size();i++)
{
tmp.push_back(candidates[i]);
dfs(ans,candidates,tmp,target-candidates[i],i+1);//与上一题不同的地方,因为不能重复选取,所以要将当前下标+1传入下一个递归。
tmp.pop_back();
while(i+1<candidates.size() && candidates[i]==candidates[i+1])//过滤相同的值,防止组合重复
i++;
}
}
}
vector<vector<int>> combinationSum2(vector<int>& candidates, int target)
{
vector<vector<int>> ans;
vector<int> tmp;
if(candidates.empty())
return ans;
sort(candidates.begin(),candidates.end());
dfs(ans,candidates,tmp,target,0);
return ans;
}
};