求解幂集问题【回溯法】

问题描述

对于给定的正整数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)可扩展出两个状态：

1. 不选择a[i]，下一状态(i+1,x[i]=0)
2. 不选择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;
}