题意
有一个专门为了集合运算而设计的“集合栈”计算机。该机器有一个开始为空的栈并且支持以下操作。 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)次操作,保证操作均能顺利进行(不需要对空栈执行出栈操作)。
输出
对于输入中指定的每个操作,输出栈顶集合的大小(即元素个数)并换行。 每组测试数据后输出一行“***”。
样例输入
2 9 PUSH DUP ADD PUSH ADD DUP ADD DUP UNION 5 PUSH PUSH ADD PUSH INTERSECT
样例输出
0 0 1 0 1 1 2 2 2 *** 0 0 1 0 0 ***
这个题主要是set模板的使用,set中有取并集和交集的模板,直接套用模板就行了
对于set<int>a,b,c;
1.set_union(a.begin(),a.end(),b.begin(),b.end(),inserter(c,c.begin())用来求两个集合的并集(去重),a是第一个集合,b是第二个集合,c是第三个集合,意思是把前两个集合取并集的结果插入到第三个集合,这样集合c就是前两个集合的并集了
2.set_intersection(5个参数同上),求两个集合的交集
完整代码:
#include <bits/stdc++.h>
#define int long long
#define into(x) inserter(x,x.begin())//因为后面用到的那两个函数参数比较多,所以可以在这里先定义一下,后面写起来就方便了
#define all(x) x.begin(),x.end()
using namespace std;
map<set<int>,int>mp;//map用来统计当前set<int>类型的集合中元素的个数
vector<set<int> >v;//vector中放题目中出现的集合
int getnum(set<int> a)//取当前集合的元素个数
{
if(mp.count(a)) return mp[a];
v.push_back(a);
mp[a]=v.size()-1;
return mp[a];
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
int t;
cin>>t;
while(t--)
{
mp.clear();
v.clear();
stack<int>t;//这里栈用来存储把每个集合用其元素个数表示后的这些数字
int n;
cin>>n;
while(n--)
{
set<int>st,x1,x2,x;
string op;
cin>>op;
if(op=="PUSH") t.push(getnum(st));//把每个集合用其元素个数表示后将这些数字依次存入栈中
else if(op=="DUP") t.push(t.top());
else{
x1=v[t.top()];t.pop();//这两句就是取前两个集合
x2=v[t.top()];t.pop();
if(op=="UNION"){
set_union(all(x1),all(x2),into(x));//取x1和x2的并集放在x,也就是说x是x1和x2的并集
}
else if(op=="INTERSECT"){
set_intersection(all(x1),all(x2),into(x));//取x1和x2的交集放在x,也就是说x是x1和x2的交集
}
else if(op=="ADD"){
x=x2;
x.insert(getnum(x1));//将x1转化为其元素个数后插入到x2中作为当前集合x
}
t.push(getnum(x));//把经过每次操作后更新的当前状态的集合x转化为其元素个数压入栈中
}
cout<<v[t.top()].size()<<endl;//最后输出栈顶元素(集合)的大小就行了
}
cout<<"***"<<endl;
}
}