The SetStack Computer集合栈计算器（stack的运用）

The SetStack Computer

   有一个专门为了集合运算而设计的“集合栈”计算机。该机器有一个初始为空的栈，并且支持以下操作：

PUSH：空集“{}”入栈
DUP：把当前栈顶元素复制一份后再入栈
UNION：出栈两个集合，然后把两者的并集入栈
INTERSECT：出栈两个集合，然后把二者的交集入栈
ADD：出栈两个集合，然后把先出栈的集合加入到后出栈的集合中，把结果入栈
每次操作后，输出栈顶集合的大小（即元素个数）。例如栈顶元素是A={ {}， {{}} }， 下一个元素是B={ {}， {{{}}} }，则：
UNION操作将得到{ {}， {{}}， {{{}}} }，输出3.
INTERSECT操作将得到{ {} }，输出1
ADD操作将得到{ {}， {{{}}}， { {}， {{}} } }，输出3.
（输入：先输入测试次数，再输入操作次数，再输入具体操作）

Sample Input

2
9
PUSH
DUP
ADD
PUSH
ADD
DUP
ADD
DUP
UNION
5
PUSH
PUSH
ADD
PUSH
INTERSECT

Sample Output

0
0
1
0
1
1
2

---

0
0
1
0
0

---

---

思路：

给每个集合分配一个编号，在栈中存储编号，注意ADD,UNION和INTERSECTION对栈顶两个集合进行操作，可以提取共性
代码分析可以看紫皮书的栈部分

---

可以来练习对STL的运用

---

#include <bits/stdc++.h>
using namespace std;
typedef set<int> Set;  //用Set表示set<int>类型
map<set<int>,int> findid;
vector<set<int> > match;
stack<int> s;

int id(Set a)   //给每个集合分配id或返回已有Id值
{
    if(findid.count(a)) return findid[a];
    match.push_back(a);
    return findid[a]=match.size()-1;
}


int main()
{
    int T;cin>>T;
    while(T--)
    {
        int n;cin>>n;
        for(int i=0;i<n;i++)
        {
            string str;
            cin>>str;
            if(str[0]=='P') s.push(id(Set()));  //Set()表示建立一个空集
            else if(str[0]=='D') s.push(s.top());
            else
            {
                Set x;
                Set x1=match[s.top()]; s.pop();
                Set x2=match[s.top()]; s.pop();
                if(str[0]=='U')
                    set_union(x1.begin(),x1.end(),x2.begin(),x2.end(),inserter(x,x.begin()));  //求并集，放入x
                else if(str[0]=='I')
                    set_intersection(x1.begin(),x1.end(),x2.begin(),x2.end(),inserter(x,x.begin()));//求交集，放入x
                else
                {
                    x2.insert(id(x1)); x=x2; //把包含x1的新集合x2赋给x
                }
                s.push(id(x));  //x统一操作
            }
            Set x=match[s.top()];
            cout<<x.size()<<endl;
        }
        cout<<"***"<<endl;
    }
    return 0;
}