问题:求含n个元素的集合的幂集。如:A={1,2,3},则A的幂集是:
p(A)={{1,2,3},{1,2},{1,3},{1},{2,3},{2},{3},{空集}};
从集合A的每个元素来看,它只有两种状态:它或者属幂集的元素集,或不属幂集的元素集。则求幂集p(A)的元素的过程可看成是依次对集合A中元素进行取舍过程,并且可以用一棵二叉树(下图),来表示过程中幂集元素的状态变化过程。其中树中的根结点表示幂集元素的初始状态(为空集),叶子结点表示它的终结状态。
集合A{1,2,3}幂集元素在生成过程中的状态图
因此求幂集元素的过程即先序遍历这棵状态树的过程。所以可以用递归和非递归两种方法求解。
方法一:递归的方法
void GetPowerSet(int i,vector<int> A,vector<int> &B) //求集合A的幂集,递归算法
{
if(i>A.size())Print(B);
else
{
int elem=A[i-1];
B.push_back(elem); //取集合A中的第i个元素
GetPowerSet(i+1,A,B);
vector<int>::iterator iter=B.end()-1;
B.erase(iter); //舍集合A中的第i个元素
GetPowerSet(i+1,A,B);
}
}
方法二:非递归的方法,整个源程序
#include <iostream>
#include<vector>
#include<iterator>
#include <stack>
using namespace std;
void Print(vector<int> B)
{
vector<int>::iterator iter;
iter=B.begin()+1;
if(iter==B.end())
{
cout<<"空集"<<endl;
return;
}
for(;iter!=B.end();iter++)
cout<<*iter<<" ";
cout<<endl;
}
void GetPowerSet(vector<int > setA)
{
stack< vector<int> > s;
vector<int> current;
int i,elem,numOfPowerSet=0;
current.push_back(0);
s.push(current);
while(!s.empty())
{
while(current[0]<setA.size())
{
i=current[0];
elem=setA[i];
current[0]+=1;
current.push_back(elem);
s.push(current);
current=s.top();
}
if(current[0]==setA.size())
{
cout<<++numOfPowerSet<<": ";
Print(current);
s.pop();
}
if(!s.empty())
{
current=s.top();
s.pop();
current[0]+=1;
s.push(current);
}
}
}
void main()
{
vector<int> A;
A.push_back(1);
A.push_back(2);
A.push_back(3);
A.push_back(4);
GetPowerSet(A);
}