组合总和
组合总和
问题描述
给定一个无重复元素的数组 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]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum
思路
本题用回溯算法,思路比较简单。
要解决的问题是,从组合中选取几个数,使得和为target。思路和以前的题没啥太大的区别,这里就不多做赘述了。
需要注意的两点是,1、没有个数的限制,只需要保证sum==target。所以,当深度达到sum>=target时,返回到上一层。这里同样需要注意,返回到上一层后将tmp.pop_back();将sum-=candidates[i];
题解
class Solution {
public:
vector<int> tmp;
vector<vector<int> > ret;
int n;
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
n=candidates.size();
dfs(candidates,target,0,0);
return ret;
}
void dfs(vector<int> &candidates,int target,int sum,int location){
if(sum>target){
return;
} else if(sum==target){
ret.push_back(tmp);
return;
}
for(int i=location;i<n;i++){
tmp.push_back(candidates[i]);
sum+=candidates[i];
dfs(candidates,target,sum,i);
tmp.pop_back();
sum=sum-candidates[i];
}
}
};