Leetcode 78 Subsets
题目原文
Given a set of distinct integers, nums, return all possible subsets (the power set).
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],
[]
]
题意分析
给一组数,输出其所有无序子数组,空子数组也考虑在内。
解法分析
本题很容易想到写出一棵子集树,用回溯的方法解,只是没有任何剪枝函数,仅仅深度优先遍历这个子集树,C++代码如下:
class Solution {
private:
vector<int> temp;
vector<vector<int>> res;
vector<int> nums;
int n;
public:
void DFS(int k){
if(k>n){
res.push_back(temp);
return;
}
DFS(++k);
k--;
temp.push_back(nums[k-1]);
DFS(++k);
temp.pop_back();
}
vector<vector<int>> subsets(vector<int>& numbers) {
nums=numbers;
n=nums.size();
DFS(1);
return res;
}
};
注意由于子集树是一个二叉树,因此DFS内部不需要用for循环遍历同层,只需要先后进行两次递归调用DFS,用于选择0和1,对应不取或取相应数字。上述算法和backtracking类似。
本题还有一个比较巧妙的方法,同样是通过那棵二叉树找到灵感。用0-2^n-1这几个数作为最终输出中的2^n个vector的序号,对序号所表示数的二进制进行检测,某一位为1代表该vector拥有对应的一个数,依次将数填入res中。C++代码如下:
class Solution {
public:
vector<vector<int>> subsets(vector<int>& nums) {
int n=nums.size();
int m=(int)pow(2.0,n);
vector<vector<int>> res(m,vector<int>());
int i,j;
for(i=0;i<n;i++){
for(j=0;j<m;j++){
if((j>>i)&1)
res[j].push_back(nums[i]);
}
}
return res;
}
};