leetcode-Combination Sum II-40

Posted 2020-08-05 0_summer

输入一个数组，选择任意个数的元素，求和等于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 };