集合一般就是回溯。
class Solution {
public:
void dfs(int index,vector<int>& nums,vector<int>&tmp){
res.push_back(tmp);
//不用递归边界,因为不会越界,是通过for循环来的,不会无限递归
for(int i = index; i < nums.size(); ++i){
tmp.push_back(nums[i]);
dfs(i+1,nums,tmp);
tmp.pop_back();
}
}
vector<vector<int>> subsets(vector<int>& nums) {
//集合一看就是回溯,空集合也要返回,因为nums元素各不相同所以没那么难
vector<int> tmp;
dfs(0,nums,tmp);
return res;
}
private:
vector<vector<int>> res;
};
二刷,回溯的很明显,来一个index记录当前遍历到第几位,在这一层的dfs里,接下来可以选的数只有index以后的
class Solution {
public:
void dfs(int index,vector<int>& nums,vector<int> tmp){
res.push_back(tmp);
//通过循环遍历的,不会越界
for(int i = index; i < nums.size(); ++i){
tmp.push_back(nums[i]);
dfs(i+1,nums,tmp);
tmp.pop_back();
}
}
vector<vector<int>> subsets(vector<int>& nums) {
//一看就是dfs回溯算法
//只能加入在自己后面的元素
vector<int> tmp;//局部变量传参就不用考虑pop了
dfs(0,nums,tmp);
return res;
}
private:
vector<vector<int>> res;
};
很简单的dfs+回溯
class Solution {
public:
void dfs(vector<int>& nums, int index){
//不用考虑越界条件,因为是通过for循环来控制的
res.push_back(tmp);
for(int i = index; i < nums.size(); ++i){
tmp.push_back(nums[i]);
dfs(nums,i+1);
tmp.pop_back();
}
}
vector<vector<int>> subsets(vector<int>& nums) {
//dfs+回溯,先将空数组加进去
//其实就是一个取或者不取的过程
dfs(nums,0);
return res;
}
private:
vector<vector<int>> res;
vector<int> tmp;
};