递归实现指数枚举,使用一些剪枝。
对枚举获得的组合,使用特殊手法对一些重复的组合去重:
class Solution {
public:
vector<int> tmp;
vector<vector<int>> res;
bool st[110] = {0};
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(),candidates.end());
dfs(0,candidates,target);
return res;
}
void dfs(int x, vector<int> &candidates, int target) {
if(x > candidates.size()) return;
if(target == 0) {
res.emplace_back(tmp);
return;
}
for(int i = x; i < candidates.size(); i++) {
const auto &num = candidates[i];
if(target >= num) {
//排除会出现重复的组合
if(i > 0 && num == candidates[i-1] && !st[i-1]) continue;
tmp.emplace_back(num);
st[i] = true;
dfs(i+1,candidates,target-num);
tmp.pop_back();
st[i] = false;
}
}
}
};
经典的 d f s dfs dfs 全排列, s t [ i ] st[i] st[i] 标记 n u m s [ i ] nums[i] nums[i] 是否被选取。总共选取 n u m s . s i z e ( ) nums.size() nums.size() 个, x x x 表示选取了几个。
class Solution {
public:
bool st[20] = {0};
vector<vector<int>> res;
vector<vector<int>> permute(vector<int>& nums) {
vector<int> tmp;
dfs(0,nums,tmp);
return res;
}
void dfs(int x, vector<int> & nums, vector<int> &tmp) {
if(x == nums.size()) {
res.emplace_back(tmp);
return;
}
for(int i = 0; i < nums.size(); i++) {
if(!st[i]) {
st[i] = true;
tmp.push_back(nums[i]);
dfs(x+1,nums,tmp);
tmp.pop_back();
st[i] = false;
}
}
}
};
全排列+组合去重(前两题结合)
class Solution {
public:
vector<vector<int>> res;
vector<int> tmp;
bool st[10] = {0};
vector<vector<int>> permuteUnique(vector<int>& nums) {
sort(nums.begin(),nums.end());
dfs(0,nums);
return res;
}
void dfs(int x, vector<int> &nums) {
if(x == nums.size()) {
res.emplace_back(tmp);
return;
}
for(int i = 0; i < nums.size(); i++) {
if(!st[i]) {
if(i > 0 && nums[i-1] == nums[i] && !st[i-1]) continue;
st[i] = true;
tmp.emplace_back(nums[i]);
dfs(x+1, nums);
tmp.pop_back();
st[i] = false;
}
}
}
};