集合栈计算机(the setstack computer, acm/icpc nwerc 2006,uva12096)

题目描述

有一个专门为了集合运算而设计的“集合栈”计算机。该机器有一个初始为空的栈，并且支持以下操作：
push：空集“{}”入栈
dup：把当前栈顶元素复制一份后再入栈
union：出栈两个集合，然后把两者的并集入栈
intersect：出栈两个集合，然后把二者的交集入栈
add：出栈两个集合，然后把先出栈的集合加入到后出栈的集合中，把结果入栈
       每次操作后，输出栈顶集合的大小（即元素个数）。例如栈顶元素是a={ {}， {{}} }， 下一个元素是b={ {}， {{{}}} }，则：
union操作将得到{ {}， {{}}， {{{}}} }，输出3.
intersect操作将得到{ {} }，输出1
add操作将得到{ {}， {{{}}}， { {}， {{}} } }，输出3.
样例输入

6
push
push
union
push
push
add

样例输出

0
0
0
0
0
1

代码实现

#define local
#include<set>
#include<stack>
#include<iostream>
#include<map>
#include<vector>
#include<algorithm> // for set_union set_intersection
using namespace std;

/*
push:空集{}入栈
union:出栈两个集合，然后把两者的并集入栈
intersect:出栈两个集合，然后把二者的交集入栈
add:出栈两个集合，然后把先出栈的集合加入到后出栈的集合中，把结果入栈。
*/
typedef set<int> set;
map<set,int> idcache; //set - id  such as:idcache[set]
vector<set> setcache; //id - cache such as:setcache[id]
stack<int> s;
int id(set x){ //光一个map数据结构还不够，这是因为有可能一个set还没有对应的id,所以这个函数应该包括的功能有:1.创建id(如果没有) 2.返回id
    if(!idcache.count(x)){
        setcache.push_back(x);
        idcache[x]=setcache.size()-1;//!!!骚操作
    }
    return idcache[x];
}
int n;
int main(){
    #ifdef local
    freopen("data.in","r",stdin);
    freopen("data.out","w",stdout);
    #endif

    cin>>n;
    while(n--){
        string exec;
        cin>>exec;
        if(exec[0] == 'p') s.push(id(set()));
        else if(exec[0] == 'd') s.push(s.top());
        else{
            set x1 = setcache[s.top()];
            s.pop();
            set x2 = setcache[s.top()];
            s.pop();
            set x;
            if(exec[0] == 'u') set_union(x1.begin(),x1.end(),x2.begin(),x2.end(),inserter(x,x.begin()));//!!!inserter
            if(exec[0] == 'i') set_intersection(x1.begin(),x1.end(),x2.begin(),x2.end(),inserter(x,x.begin()));//!!!inserter
            if(exec[0] == 'a'){
                x=x2;
                x.insert(id(x1));
            }
            s.push(id(x));
        }
        cout<<setcache[s.top()].size()<<endl;
    }
}