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

Combination Sum

时间:2015-07-13 22:18:47      阅读:95      评论:0      收藏:0      [点我收藏+]

标签:

题目描述:
给定一个包含多个正数的set容器和目标值target,从set中找出sum等于target的组合,同样的数可以使用多次。例如 set 为 [2,3,6,7]并且target为7,结果为 [7] [2,2,3].

分析:我们先将集合排序,将sum初始化为0,从最小的数开始,将其加到sum上,并存入容器mid保存,将sum与target比较,小于 target递归此操作,等于target则将mid加入结果容器,大于target则返回,从mid容器中退出该数,将sum减去该数;对下一个数进行 同样的操作。具体代码如下:

 1 void dfscombine(vector<int>& candidates,int level,int& sum,int target,vector<int>& mid,vector<vector<int> >& result)
 2     {
 3         if(sum>target) return;
 4         else if(sum==target)
 5            result.push_back(mid);
 6         else
 7         {
 8           for(int i=level;i<candidates.size();i++)
 9           {
10               sum+=candidates[i];
11               mid.push_back(candidates[i]);
12               dfscombine(candidates,i,sum,target,mid,result);
13               mid.pop_back();
14               sum-=candidates[i];
15           }
16         }
17     }
18     vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
19         vector<int> mid;  
20         vector<vector<int> > result;  
21         sort(candidates.begin(),candidates.end());
22         int level=0,sum=0;
23         dfscombine(candidates,level,sum,target,mid,result);
24         return result;
25     }

上题中允许一个数使用多次,现在要求每个数只能使用一次,并且上题中的set该为collision,这便成了Combination Sum II。 从dfscombine(candidates,i,sum,target,mid,result)这一行代码中可以看出,我们每次递归都candidates[i]重复计算,为了保证每个数只出现一次,只需将其中的i该为i+1即可,因为collision中会存在相等的数,为了保证结果的唯一性要在for循环最后加入代码:

 

while(i<candidates.size()-1 && candidates[i]== candidates[i+1]) i++; 

 

Combination Sum III

在1到9中找到k个数,使其和等于目标值。例如k=3时,目标值为9时,有结果:[[1,2,6], [1,3,5], [2,3,4]]
该题解法与上述两天类似,判断时当mid容器中元素个数大于k也要返回,当和值等于目标值并且mid容器中元素个数等于k时才能加入结果容器,代码如下:

 1 void dfscombine(vector<int>& candidates,int level,int& sum,int target,vector<int>& mid,vector<vector<int> >& result,int k)
 2     {
 3         if(sum>target||mid.size()>k) return;
 4         else if(sum==target&&mid.size()==k)
 5            result.push_back(mid);
 6         else
 7         {
 8           for(int i=level;i<candidates.size();i++)
 9           {
10               sum+=candidates[i];
11               mid.push_back(candidates[i]);
12               dfscombine(candidates,i+1,sum,target,mid,result,k);
13               mid.pop_back();
14               sum-=candidates[i];
15           }
16         }
17     }
18     vector<vector<int>> combinationSum3(int k, int n) {
19         vector<int> mid;  
20         vector<vector<int> > result; 
21         vector<int> a(9,0);
22         for(int i=1;i<10;i++)
23            a[i-1]=i;
24         int level=0,sum=0;
25         dfscombine(a,level,sum,n,mid,result,k);
26         return result;
27     }

 

Combination Sum

标签:

原文地址:http://www.cnblogs.com/zhulong890816/p/4643903.html

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