Given a set of distinct integers, S, return all possible subsets.
Note:
Elements in a subset must be in non-descending order.
The solution set must not contain duplicate subsets.
For example,
If S = [1,2,3], a solution is:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
非递归解法:
思路:
拿给的例子[1,2,3]来说,最开始是空集,那么我们现在要处理1,就在空集上加1,为[1],现在我们有两个子集[]和[1],接下来在之前的子集基础上,每个都加个2,可以分别得到[2],[1, 2],那么现在所有的子集合为[], [1], [2], [1, 2],同理处理3的情况可得[3], [1, 3], [2, 3], [1, 2, 3], 再加上之前的子集就是所有的子集合了,代码如下:
#include <iostream>
#include <vector>
using namespace std;
vector<int> input;
vector<vector<int>> ans;
vector<vector<int> > subsets(vector<int> &S) {
vector<vector<int>> res(1);//初始化为1个元素,空
sort(S.begin(),S.end());//要求非递减
for(int i=0;i<S.size();i++){
int size=res.size();
for(int j=0;j<size;j++){
res.push_back(res[j]);//加入上一轮已有的元素
res.back().push_back(S[i]);//新加入一个元素
}
}
return res;
}
int main(){
int n,x;
cin>>n;
for(int i=0;i<n;i++){
cin>>x;
input.push_back(x);
}
ans=subsets(input);
for(int i=0;i<ans.size();i++){
cout<<"[";
for(int j=0;j<ans[i].size();j++){
cout<<ans[i][j];
}
cout<<"]"<<endl;
}
}
input:
3
1 2 3
output:
[]
[1]
[2]
[12]
[3]
[13]
[23]
[123]
递归解法,类似于dfs:
思路:
由于原集合每一个数字只有两种状态,要么存在,要么不存在,那么在构造子集时就有选择和不选择两种情况,所以可以构造一棵二叉树,左子树表示选择该层处理的节点,右子树表示不选择,最终的叶节点就是所有子集合,树的结构如下:
[]
/ \
/ \
/ \
[1] []
/ \ / \
/ \ / \
[1 2] [1] [2] []
/ \ / \ / \ / \
[1 2 3] [1 2] [1 3] [1] [2 3] [2] [3] []
vector<vector<int> > res;
vector<int> temp;
void getSubsets(vector<int> &S, int pos, vector<int> &temp, vector<vector<int> > &res) {
res.push_back(temp);
for (int i = pos; i < S.size(); ++i) {
temp.push_back(S[i]);
getSubsets(S, i + 1, temp, res);
temp.pop_back();//删除最后一个元素pop_back()
}
}
vector<vector<int> > subsets(vector<int> &S) {
sort(S.begin(), S.end());
getSubsets(S, 0, temp, res);
return res;
}
牛客网上的这道题,要求按照元素个数非递减排序输出
void getSubsets(vector<int> &S, int pos,int k) {
if(k<0) return;
else if(k==0)
{
res.push_back(temp);
}
else{
for (int i = pos; i < S.size(); ++i) {
temp.push_back(S[i]);
getSubsets(S, i + 1,k-1);
temp.pop_back();//删除最后一个元素pop_back()
}
}
}
vector<vector<int> > subsets(vector<int> &S) {
sort(S.begin(), S.end());
for(int i=0;i<=S.size();i++){
getSubsets(S, 0, i);}//用i控制元素个数,保证有序
return res;
}