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

#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");
    }
}