问题描述
对于给定的正整数n,求1~n构成的集合的幂集。(即由1-n集合中所有子集构成的集合,包括全集和空集。
例如 a[3]={1, 2, 3}.幂集为:{ }, {1}, {2}, {3}, {1 2}, {1 3}, {2 3}, {1 2 3}
回溯法
本问题的解空间为子集树,每个元素只有两种扩展,要么选择,要么不选择。采用深度优先搜索思路,解向量为x[],x[i]=0表示不选择a[i],x[i]=1表示选择a[i]。初始:i=0,x[]=0->目标:i=n,x为一个解。从状态(i,x)可扩展出两个状态:
- 不选择a[i],下一状态(i+1,x[i]=0)
- 不选择a[i],下一状态(i+1,x[i]=1)
因为i递增,不会出现重复情况。
#include<bits/stdc++.h>
using namespace std;
int x[10];
//输出一个解
void PrintA(int a[], int n, int x[])
{
cout<<" {";
for(int i=0; i<n; i++)
if(x[i]==1) cout<<a[i];
cout<<"}";
}
//回溯法求解向量x
void dfs(int a[], int n, int i, int x[])
{
if(i>=n) PrintA(a, n, x);
else
{
x[i]=0; dfs(a, n, i+1, x); //不选择a[i]
x[i]=1; dfs(a, n, i+1, x); //选择a[i]
}
}
int main()
{
int a[]={1, 2, 3};
int n=sizeof(a)/sizeof(*a);
cout<<"幂集为\n";
dfs(a, n, 0, x);
cout<<endl;
}
vector改进
用一个path容器存放获取的子集。
#include<bits/stdc++.h>
using namespace std;
//输出一个解
void PrintA(vector<int> path)
{
cout<<" {";
for(int i=0; i<path.size(); i++)
cout<<path[i];
cout<<"}";
}
//回溯法求解子集path
void dfs(int a[], int n, int i, vector<int> path)
{
if(i>=n) PrintA(path);
else
{
dfs(a, n, i+1, path); //不选择a[i]
path.push_back(a[i]);
dfs(a, n, i+1, path); //选择a[i]
}
}
int main()
{
int a[]={1, 2, 3};
int n=sizeof(a)/sizeof(*a);
vector<int> path;
cout<<"幂集为\n";
dfs(a, n, 0, path);
cout<<endl;
}