LintCode-數字組合 II
阿新 • • 發佈:2019-01-24
給出一組候選數字(C)和目標數字(T),找出C中所有的組合,使組合中數字的和為T。C中每個數字在每個組合中只能使用一次。
樣例給出一個例子,候選數字集合為[10,1,6,7,2,1,5] 和目標數字 8,
解集為:[[1,7],[1,2,5],[2,6],[1,1,6]]
注意- 所有的數字(包括目標數字)均為正整數。
- 元素組合(a1, a2,
… , ak)必須是非降序(ie, a1 ≤ a2 ≤
… ≤ ak)。
- 解集不能包含重複的組合。
分析:先排序,在對每個元素採取取或者不取的遞迴操作。
程式碼:
class Solution { public: /** * @param num: Given the candidate numbers * @param target: Given the target number * @return: All the combinations that sum to target */ vector<vector<int> > combinationSum2(vector<int> &num, int target) { // write your code here set<vector<int> > r; vector<int> cur; sort(num.begin(),num.end()); combination(cur,num,0,0,target,r); vector<vector<int> > ret; copy(r.begin(),r.end(),back_inserter(ret)); return ret; } void combination(vector<int> cur,vector<int> &num,int index,int curSum,int target,set<vector<int> >&ret) { if(curSum>target) return; if(curSum==target) { ret.insert(cur); return; } if(index==num.size()) return; combination(cur,num,index+1,curSum,target,ret); cur.push_back(num[index]); combination(cur,num,index+1,curSum+num[index],target,ret); } };