1. 程式人生 > >LintCode-數字組合 II

LintCode-數字組合 II

給出一組候選數字(C)和目標數字(T),找出C中所有的組合,使組合中數字的和為T。C中每個數字在每個組合中只能使用一次。

樣例

給出一個例子,候選數字集合為[10,1,6,7,2,1,5] 和目標數字 8,

解集為:[[1,7],[1,2,5],[2,6],[1,1,6]]

注意
  • 所有的數字(包括目標數字)均為正整數。
  • 元素組合(a1a2, … , 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);
    }
};