白书的例题,而且提示十分重要,所以就做了一下。
先说题意
维护 N(1≤N≤2000) 个操作, 可能的五种操作如下: PUSH: 在栈顶加入一个空集合 A={} ; DUP: 把栈顶集合 A 复制一遍再加进去; UNION: 弹出栈顶集合 A 及其下集合 B , 把它们的并集加入栈; INTERSECT: 弹出栈顶集合 A 及其下集合 B , 把他们的交集加入栈; ADD: 弹出栈顶集合 A 及其下集合 B , 把栈顶集合 A 作为一个元素加入集合 B , 再把集合 B 加入栈中.
对于每个操作, 输出栈顶集合 A 中的元素个数.
然后白书就用了map,vector,set和stack,当然stack和set是一定要用的,至于为什么用map和vector就大有讲究了,这个题在用栈来存储集合的集合,因此主要思路就是用数字为集合编号,每一个集合对应一个编号,这样栈里的元素就是数字了,这个主要是针对最后一个ADD来用的,因为如果不用编号的话,很难把一个集合当成一个元素放到另一个集合里(想想其实也可以,用set< set<int> >这样的方法可能会行)。所以就像课本那样了。不说了,上代码吧,有不懂的,请看白书第二版P116.
#include<iostream>
#include<map>
#include<set>
#include<stack>
#include<vector>
#include<algorithm>
#define ALL(x) x.begin(),x.end()
#define INS(x) inserter(x,x.begin())
using namespace std;
typedef set<int> Set;
vector<Set> Setcache;
map<Set,int> IDcache;
int ID(Set s) {
if(IDcache.count(s)) return IDcache[s];
Setcache.push_back(s);
return IDcache[s]=Setcache.size()-1;
}
int main() {
stack<int> s;
int n,m;
cin>>m;
while(m--) {
cin>>n;
for(int i=0; i<n; i++) {
string op;
cin>>op;
if(op[0] == 'P') s.push( ID(Set()) );
else if(op[0] == 'D') s.push(s.top());
else {
Set x1=Setcache[s.top()];
s.pop();
Set x2=Setcache[s.top()];
s.pop();
Set x;
if(op[0] == 'U') set_union(ALL(x1),ALL(x2),INS(x));
if(op[0] == 'I') set_intersection(ALL(x1),ALL(x2),INS(x));
//把代表集合x1的编号插入到x中
if(op[0] == 'A') {
x=x2;
x.insert( ID(x1) );
}
s.push( ID(x) );
}
cout<<Setcache[s.top()].size()<<endl;
}
cout<<"***"<<endl;
}
return 0;
}