题意:
有一个专门为了集合运算而设计的“集合栈”计算机。该机器有一个开始为空的栈并且支持以下操作。
1.PUSH:空集“{}”入栈。
2.DUP:把当前栈顶元素复制一份后在入栈。
3.UNION:出栈两个集合,然后把二者的并集入栈。
4.INTERSECT:出栈两个集合,然后把二者的交集入栈。
5.ADD:出栈两个集合,然后把先出栈的集合加入到后出栈的集合中,把结果入栈。
例如,栈顶元素是A = { {},{{}} },下一个元素是B = { {}, {{{}}} },则:
UNION 操作将得到{ {}, {{}}, {{{}}} },输出3.
INTERSECT 操作将得到{ {} },输出1。
1.PUSH:空集“{}”入栈。
2.DUP:把当前栈顶元素复制一份后在入栈。
3.UNION:出栈两个集合,然后把二者的并集入栈。
4.INTERSECT:出栈两个集合,然后把二者的交集入栈。
5.ADD:出栈两个集合,然后把先出栈的集合加入到后出栈的集合中,把结果入栈。
例如,栈顶元素是A = { {},{{}} },下一个元素是B = { {}, {{{}}} },则:
UNION 操作将得到{ {}, {{}}, {{{}}} },输出3.
INTERSECT 操作将得到{ {} },输出1。
ADD 操作将得到{ {}, {{{}}},{{} , {{}}} },输出3。
第一行有一个整数T(0<=T<=5)表示有T组测试数据。
每组测试数据有N(0<=N<=2000)次操作,保证操作均能顺利进行(不需要对空栈执行出栈操作)。
对于输入中指定的每个操作,输出栈顶集合的大小(即元素个数)并换行。
每组测试数据后输出一行“***”。
思路:膜拜刘汝佳大神。我们可以用map使得每个不同set对应一个唯一的ID(int),且用vector使得这个ID(vector的下标)对应一个set。这样我们开一个栈去存储ID就可以模拟上述五种操作。
#include<bits/stdc++.h>
using namespace std;
typedef set<int> Set;
map<Set, int> IDcache;
vector<Set> Setcache;
#define ALL(x) x.begin(),x.end()
#define INS(x) inserter(x,x.begin())
//查找指定集合的ID。如果找不到,分配一个新的ID
int ID(Set x)
{
if(IDcache.count(x)) return IDcache[x];
Setcache.push_back(x);//添加新集合
return IDcache[x] = Setcache.size() - 1;
}
int main()
{
int T;
scanf("%d", &T);
while (T--)
{
int N;
scanf("%d", &N);
stack<int> s;//放每个元素的ID
while (N--)
{
string op;
cin >> op;
if (op[0] == 'P')
{
s.push(ID(Set()));//Set() == set<int> () 意为一个空集和
}
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));//x1和x2取并集,然后放入x
else if(op[0] == 'I') set_intersection(ALL(x1), ALL(x2), INS(x));//x1和x2取交集,然后放入x
else if(op[0] == 'A') { x = x2; x.insert(ID(x1)); }//A作为一个整体放入x2
s.push(ID(x));//x的ID放入栈
}
printf("%d\n",Setcache[s.top()].size());
}
printf("***\n");
}
return 0;
}