Given a set of distinct integers, nums, return all possible subsets.
Note: The solution set must not contain duplicate subsets.
For example,
If nums = [1,2,3], a solution is:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
分析:求集合的所有子集问题。题目要求子集中元素非递减序排列,因此我们先要对原来的集合进行排序。原集合中每一个元素在子集中有两种状态:要么存在、要么不存在。这样构造子集的过程中每个元素就有两种选择方法:选择、不选择,因此可以构造一颗二叉树,例如对于例子中给的集合[1,2,3],构造的二叉树如下(左子树表示选择该层处理的元素,右子树不选择),最后得到的叶子节点就是子集:
class Solution {
private:
vector<vector<int>> res;
public:
vector<vector<int>> subsets(vector<int>& nums) {
res.clear();
sort(nums.begin(),nums.end());
vector<int> tmp;
dfs(nums,0,tmp);
return res;
}
private:
void dfs(vector<int> nums,int index,vector<int>& tmp){
if(index == nums.size()){
res.push_back(tmp);
return;
}
tmp.push_back(nums[index]);
dfs(nums,index+1,tmp);
tmp.pop_back();
dfs(nums,index+1,tmp);
}
};
类似的解法:以{1,2,3}为例,首先取出1,加入子集合,后面的2和3就是1的子节点,先取出2,把[1, 2]加入子集合,后面的3就是2的子节点,取出3,把[1, 2, 3]加入子集合。然后回溯,取出3,将[1, 3]加入子集合。
1处理完成之后,我们可以同样方式处理2,以及3。
class Solution {
public:
vector<vector<int> > res;
vector<vector<int> > subsets(vector<int> &S) {
res.clear();
vector<int> tmp;
//sort(S.begin(),S.end());
res.push_back(vector<int>());
dfs(S,0,tmp);
return res;
}
void dfs(vector<int> nums, int start, vector<int>& tmp){
if(start == nums.size()){
return ;
}
for(int i = start; i<nums.size(); i++){
tmp.push_back(nums[i]);
res.push_back(tmp);
dfs(nums,i+1,tmp);
tmp.pop_back();
}
}
};
subsets Ⅱ 是在subsetⅠ的基础上去重。思想类似。
class Solution {
public:
vector<vector<int>> res;
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
res.clear();
sort(nums.begin(),nums.end());
res.push_back(vector<int>());
vector<int> tmp;
dfs(nums,0,tmp);
return res;
}
void dfs(vector<int> nums, int start, vector<int> &tmp){
if(start == nums.size()){
return ;
}
vector<vector<int>>::iterator it = res.begin();
for(int i = start; i<nums.size(); i++){
tmp.push_back(nums[i]);
res.push_back(tmp);
dfs(nums,i+1,tmp);
tmp.pop_back();
while(i<nums.size()-1 && nums[i]==nums[i+1]){
i++;
}
}
}
};