描述
给出一个候选数字的set(C)和目标数字(T),找到C中所有的组合,使找出的数字和为T。C中的数字可以无限制重复被选取。
例如,给出候选数组[2,3,6,7]和目标数字7,所求的解为:
[7],
[2,2,3]
所有的数字(包括目标数字)均为正整数。
元素组合(a1, a2, … , ak)必须是非降序(ie, a1 ≤ a2 ≤ … ≤ ak)。
解集不能包含重复的组合。
样例
给出候选set[2,3,6,7]和目标数字7
给出一个候选数字的set(C)和目标数字(T),找到C中所有的组合,使找出的数字和为T。C中的数字可以无限制重复被选取。
例如,给出候选数组[2,3,6,7]和目标数字7,所求的解为:
[7],
[2,2,3]
所有的数字(包括目标数字)均为正整数。
元素组合(a1, a2, … , ak)必须是非降序(ie, a1 ≤ a2 ≤ … ≤ ak)。
解集不能包含重复的组合。
样例
给出候选set[2,3,6,7]和目标数字7
返回 [[7],[2,2,3]]
分析
该题目可以使用回溯法,对每加入一个数与目标进行判断,如果小于则再加入,如果大于,则跳下一个,如果等于,则存到返回数组里面。
程序
class Solution {
public:
/**
* @param candidates: A list of integers
* @param target: An integer
* @return: A list of lists of integers
*/
// 分治思想
vector<vector<int>> combinationSum(vector<int> &candidates, int target) {
// write your code here
vector<vector<int> > results;
vector<int> result;
int sum = 0;
sort(candidates.begin(), candidates.end());
combinationSum(results, result, candidates, target, sum, 0);
return results;
}
void combinationSum(vector<vector<int> > &results, vector<int> &result, vector<int> &candidates, int target, int sum, int level){
if(sum > target || level >= candidates.size())
return;
if(sum == target){
results.push_back(result);
return;
}
for(int i = level; i < candidates.size(); i++){
sum += candidates[i];
result.push_back(candidates[i]);
combinationSum(results, result, candidates, target, sum, i);
result.pop_back();
sum -= candidates[i];
}
}
};