题目:78. 子集
方法一:枚举每一个数选还是不选,深度优先搜索dfs。
C++版本:
class Solution {
public:
vector<vector<int>> v;
vector<int> tmp;
void dfs(vector<int> & nums,int u){
if(u==nums.size()){
v.push_back(tmp);
return ;
}
dfs(nums,u+1);
tmp.push_back(nums[u]);
dfs(nums,u+1);
tmp.pop_back();
}
vector<vector<int>> subsets(vector<int>& nums) {
dfs(nums,0);
return v;
}
};
JAVA版本:
class Solution {
List<List<Integer>> v;
List<Integer> tmp;
void dfs(int[] nums,int u){
if(u==nums.length){
v.add(new ArrayList<>(tmp));
return;
}
dfs(nums,u+1);
tmp.add(nums[u]);
dfs(nums,u+1);
tmp.remove(tmp.size()-1);
}
public List<List<Integer>> subsets(int[] nums) {
v=new ArrayList<>();
tmp=new ArrayList<>();
dfs(nums,0);
return v;
}
}
方法二:枚举下一个位置插入的数,深度优先搜索dfs。
C++版本:
class Solution {
public:
vector<vector<int>> v;
vector<int> tmp;
void dfs(vector<int> & nums,int u){
v.push_back(tmp);
for(int i=u;i<nums.size();i++){
tmp.push_back(nums[i]);
dfs(nums,i+1);
tmp.pop_back();
}
}
vector<vector<int>> subsets(vector<int>& nums) {
dfs(nums,0);
return v;
}
};
JAVA版本:
class Solution {
List<List<Integer>> v;
List<Integer> tmp;
void dfs(int[] nums,int u){
v.add(new ArrayList<>(tmp));
for(int i=u;i<nums.length;i++){
tmp.add(nums[i]);
dfs(nums,i+1);
tmp.remove(tmp.size()-1);
}
}
public List<List<Integer>> subsets(int[] nums) {
v=new ArrayList<>();
tmp=new ArrayList<>();
dfs(nums,0);
return v;
}
}
方法三:位运算。
C++版本:
class Solution {
public:
vector<vector<int>> subsets(vector<int>& nums) {
vector<vector<int>> v;
int n=nums.size();
for(int i=0;i<(1<<n);i++){
vector<int> tmp;
for(int j=0;j<n;j++){
if((i>>j)&1==1){
tmp.push_back(nums[j]);
}
}
v.push_back(tmp);
}
return v;
}
};
JAVA版本:
class Solution {
public List<List<Integer>> subsets(int[] nums) {
List<List<Integer>> v=new ArrayList<>();
int n=nums.length;
for(int i=0;i<(1<<n);i++){
List<Integer> tmp=new ArrayList<>();
for(int j=0;j<n;j++){
if(((i>>j)&1)==1){
tmp.add(nums[j]);
}
}
v.add(tmp);
}
return v;
}
}