UVa - 12096 集合栈计算机（STL）

Posted 2020-11-02 jvruodejrLS

【题意】

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

其他操作输出0

 

多组输入输出，每组输出后输出一行“ *** ”

输入不超过2000个操作

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
2
2
***
0
0
1
0
0
***

【代码】

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <algorithm>
using namespace std;

#define ALL(x) x.begin(), x.end()
#define INS(x) inserter(x, x.begin())
//感觉这两个宏用的真的厉害

typedef set<int> Set;
map<Set, int> IDcache;
vector<Set> Setcache;

int ID(Set x) //找到集合x在列表中的代表id
{
    if (IDcache.count(x)) return IDcache[x];
    Setcache.push_back(x);//如果没找到就新加一个
    return IDcache[x] = Setcache.size()-1;
}

int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        IDcache.clear();
        Setcache.clear();

        stack<int> s;
        int n;
        scanf("%d", &n);
        for (int i = 1; i <= n; i++)
        {
            string opr;
            cin >> opr;
            if (opr[0] == ‘P‘)
                s.push(ID(Set()));
            else if (opr[0] == ‘D‘)
                s.push(s.top());
            else
            {
                Set x, y, sum;
                x = Setcache[s.top()]; s.pop();
                y = Setcache[s.top()]; s.pop();

                if (opr[0] == ‘U‘)
                    set_union(ALL(x), ALL(y), INS(sum));
　　　　　　　　　　　　//set_union()合并x和y 存入sum
                else if (opr[0] == ‘I‘)
                    set_intersection(ALL(x), ALL(y), INS(sum));
　　　　　　　　　　　　//set_intersection()求x和y的交集，存入sum
                else if (opr[0] == ‘A‘)
                {
                    sum = y;
                    sum.insert(ID(x));//不是把两个集合合并，而是x是y的一个元素
                }
                s.push(ID(sum));
            }
            printf("%d\n", Setcache[s.top()].size());
        }
        printf("***\n");
    }
}