leetcode 子集 深搜
一(每一个数选或不选
给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: nums = [1,2,3]
输出:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subsets
class Solution {
public:
vector<vector<int>> ans;
vector<int> now;
vector<vector<int>> subsets(vector<int>& nums) {
dfs(nums,0);
return ans;
}
void dfs(vector<int>& nums,int u)
{
if(u==nums.size()){
ans.push_back(now);
return;
}
//选
now.push_back(nums[u]);
dfs(nums,u+1);
now.pop_back();
dfs(nums,u+1);
}
};
二(每一个数选零到k个
给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subsets-ii
class Solution {
public:
vector<vector<int>> ans;
vector<int> now;
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(),nums.end());
dfs(nums,0);
return ans;
}
void dfs(vector<int>& nums,int u)
{
if(u==nums.size()){
ans.push_back(now);
return;
}
int k=u;
while(k<nums.size()&&nums[k]==nums[u])k++;
//不选
dfs(nums,k);
//选K个
for(int i=u;i<k;++i){
now.push_back(nums[i]);
dfs(nums,k);
}
//恢复
for(int i=u;i<k;++i)now.pop_back();
}
};