集合栈计算机

                                                                                               集合栈计算机

题目是这样的：

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

***

AC代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
#include<set>
#include<map>
#include<vector>
#include<stack>
using namespace std;
typedef set<int> Set;
map<Set,int> mp;//将每个集合编号
vector<Set> vec;//保存集合
stack<int> a;//集合在vector的位置
int ID(Set x)//查找集合的编号
{
	if(mp.count(x))return mp[x];//已经存在，直接返回
	vec.push_back(x);//新集合进栈
	return mp[x]=vec.size()-1;//返回集合在栈中的位置
}
int main()
{
	/*freopen("input.txt","r",stdin);*/
	int n,m;
	char s[10];
	cin >> n;
	while(n--)
	{
		cin >> m;
		while(m--)
		{
			cin >> s;
			if(s[0]=='P')//进一个空栈
			{
				a.push(ID(Set()));
			}
			else if(s[0]=='D')
			{
				a.push(a.top());
			}
			else
			{
				Set x1 = vec[a.top()];a.pop();
				Set x2 = vec[a.top()];a.pop();
				Set x;
				if(s[0]=='A'){
				x=x2;
				x.insert(ID(x1));
				}
				if(s[0]=='U')
		set_union(x1.begin(),x1.end(),x2.begin(),x2.end(),inserter(x,x.begin()));
				if(s[0]=='I')
		set_intersection(x1.begin(),x1.end(),x2.begin(),x2.end(),inserter(x,x.begin()));
				a.push(ID(x));
			}
			cout << vec[a.top()].size() << endl;
		}
		cout << "***\n";
	}
	return 0;
}