码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode之Combination Sum

时间:2014-08-22 16:24:19      阅读:202      评论:0      收藏:0      [点我收藏+]

标签:combination sum   递归   leetcode   

Combination Sum

Given a set of candidate numbers (C) and a target number (T), find all unique combinations in C where the candidate numbers sums to T.

The same repeated number may be chosen from C unlimited number of times.

Note:

  • All numbers (including target) will be positive integers.
  • Elements in a combination (a1a2, … , ak) must be in non-descending order. (ie, a1 ≤ a2 ≤ … ≤ ak).
  • The solution set must not contain duplicate combinations.

For example, given candidate set 2,3,6,7 and target 7
A solution set is: 
[7] 
[2, 2, 3] 

思路:典型的递归程序,对每次遍历,取当前数和不取当前数进行遍历。

class Solution {
public:
    void combinationSum(vector<int>& candidates,int index,int currentSum,int target,vector<int>& path)
    {
    	if(currentSum == target)
    	{
    		res.push_back(path);
    		return;
    	}
    	int length = candidates.size();
    	if(index == length || currentSum > target)return;
    	path.push_back(candidates[index]);
    	combinationSum(candidates,index,currentSum+candidates[index],target,path);//重复添加当前节点
    	path.pop_back();
    	combinationSum(candidates,index+1,currentSum,target,path);//跳过当前节点
    }
    
    vector<vector<int> > combinationSum(vector<int> &candidates, int target) {
        sort(candidates.begin(),candidates.end());
    	vector<int> path;
    	combinationSum(candidates,0,0,target,path);
    	return res;
    }

private:
    vector<vector<int> > res;
};

Combination Sum II

 

Given a collection of candidate numbers (C) and a target number (T), find all unique combinations in C where the candidate numbers sums to T.

Each number in C may only be used once in the combination.

Note:

  • All numbers (including target) will be positive integers.
  • Elements in a combination (a1a2, … , ak) must be in non-descending order. (ie, a1 ≤ a2 ≤ … ≤ ak).
  • The solution set must not contain duplicate combinations.

For example, given candidate set 10,1,2,7,6,1,5 and target 8
A solution set is: 
[1, 7] 
[1, 2, 5] 
[2, 6] 
[1, 1, 6] 

class Solution {
public:
    void combinationSum2(vector<int>& num,int index,int currentSum,int target,vector<int>& path)
    {
    	if(currentSum == target)
    	{
    		hash.insert(path);//防止重复
    		return;
    	}
    	int length = num.size();
    	if(index == length || currentSum > target)return;
    	path.push_back(num[index]);
    	combinationSum2(num,index+1,currentSum+num[index],target,path);//取当前节点
    	path.pop_back();
    	combinationSum2(num,index+1,currentSum,target,path);//不取当前节点
    }
    
    vector<vector<int> > combinationSum2(vector<int> &num, int target) {
    	sort(num.begin(),num.end());
    	vector<int> path;
    	combinationSum2(num,0,0,target,path);
    	vector<vector<int> > res(hash.begin(),hash.end());
    	return res;
    }
private:
    set<vector<int> > hash;
};

如果是求总数,而且不需要去除重复,可以看这里



leetcode之Combination Sum

标签:combination sum   递归   leetcode   

原文地址:http://blog.csdn.net/fangjian1204/article/details/38756033

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!