-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmain.cpp
88 lines (84 loc) · 2.7 KB
/
main.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;
//只需要在同一层DFS的时候检查相邻元素是否相同,
//避免从相同元素在同一层开始DFS就可以
class Solution {
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(), candidates.end());
vector<vector<int>> res;
vector<int> combination;
dfs(target, 0, candidates, combination, res);
return res;
}
void dfs(int target, int start, vector<int>& candidates, vector<int>& combination, vector<vector<int>>& res)
{
if(!target)
{
res.push_back(combination);
return ;
}
for(int i = start; i < candidates.size() && target >= candidates[i]; ++i)
{
if(i && candidates[i] == candidates[i-1] && i > start)
continue; // 去重,相同元素第二个进行DFS时直接跳过
combination.push_back(candidates[i]);
dfs(target - candidates[i], i + 1, candidates, combination, res);
combination.pop_back();
}
}
};
// 调试版
/*
class Solution {
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(), candidates.end());
vector<vector<int>> res;
vector<int> combination;
dfs(target, 0, candidates, combination, res);
return res;
}
void dfs(int target, int start, vector<int>& candidates, vector<int>& combination, vector<vector<int>>& res)
{
if(!target)
{
res.push_back(combination);
return ;
}
for(int i = start; i < candidates.size() && target >= candidates[i]; ++i)
{
cout << "当前的索引是:start:" << start << " " << "target:" << target<< " "<< "i:" << i <<endl;
if(i && candidates[i] == candidates[i-1] && i > start)
{
cout << "去重" << endl;
continue; // 去重,相同元素第二个进行DFS时直接跳过
}
cout << "candi:" << candidates[i] << endl;
combination.push_back(candidates[i]);
cout << "com:";
for(int i = 0; i < combination.size(); ++i)
cout << combination[i] << " ";
cout << endl;
dfs(target - candidates[i], i + 1, candidates, combination, res);
combination.pop_back();
}
}
};
*/
int main()
{
Solution s;
vector<int> data = {1,5,1};
int target = 6;
vector<vector<int>> res;
res = s.combinationSum2(data, target);
for(int i = 0; i < res.size(); ++i)
for(int j = 0; j < res[i].size(); ++j)
cout << res[i][j] << " ";
cout << endl;
return 0;
}