输入一个数组,选择任意个数的元素,求和等于target的所有组合,每个元素只能选择一次,组合不能重复
求组合:先排序,dfs,回溯
去重:把dfs想象成一颗树,同一层如果元素值相同,那么只对第一个元素深搜,其余相同的元素不必深搜,注意是同一层,看代码注释
1 class Solution { 2 public: 3 void dfs(vector<vector<int> > &v,vector<int> vv,vector<int> a,int i,int target,int sum){ 4 if(sum>target) return; 5 if(sum==target){ 6 v.push_back(vv); 7 return; 8 } 9 for(int j=i+1;j<a.size();j++){ 10 if(sum+a[j]<=target){ 11 if(j!=i+1&&a[j]==a[j-1]) continue; 12 vv.push_back(a[j]); 13 dfs(v,vv,a,j,target,sum+a[j]); 14 vv.pop_back(); 15 } 16 } 17 } 18 vector<vector<int> > combinationSum2(vector<int>& candidates, int target) { 19 vector<vector<int> > v; 20 vector<int> vv; 21 int sum=0; 22 sort(candidates.begin(),candidates.end()); 23 for(int i=0;i<candidates.size();i++){ 24 if(i>0&&candidates[i]==candidates[i-1]) continue; 25 if(candidates[i]<=target){ 26 vv.push_back(candidates[i]); 27 dfs(v,vv,candidates,i,target,sum+candidates[i]); 28 vv.pop_back(); 29 } 30 } 31 return v; 32 // if(v.size()==0) return v; 33 // vector<vector<int> > ans; 34 // ans.push_back(v[0]); 35 // int k=0; 36 // for(int i=1;i<v.size();i++){ 37 // if(v[i].size()!=ans[k].size()){ 38 // ans.push_back(v[i]); 39 // k++; 40 // continue; 41 // } 42 // int ok=0; 43 // for(int j=0;j<v[i].size();j++){ 44 // if(v[i][j]!=ans[k][j]){ 45 // ok=1;break; 46 // } 47 // } 48 // if(ok){ 49 // ans.push_back(v[i]); 50 // k++; 51 // } 52 // } 53 // return ans; 54 } 55 };
时间: 2024-10-06 03:16:10