The Setstack Computer UVa12096

本题的思想很重要,题意中的集合概念较抽象,因此对每种集合编号。

stack为题意所描述的stack computer,存放着抽象集合对应的唯一编号,而编号所对应的集合里面也存放着集合的编号,形成了抽象集合的嵌套关系。

函数getID(set):set ==> 编号,通过映射ID将每个集合对应唯一编号,以此保证集合元素的互异性,接口返回编号或者分配新的编号。

向量getSet:编号 ==> set,通过编号找到对应set,获取集合大小。

#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
map, int> ID;
vector> getSet;

int getID(set x)
{
    if (ID.count(x))
        return ID[x];
    getSet.push_back(x);
    return ID[x] = getSet.size() - 1;
}
int main()
{
    int T;
    cin >> T;
    while (T--)
    {
        stack s;
        int n;
        cin >> n;
        while (n--)
        {
            char cmd[10];
            cin >> cmd;
            if (cmd[0] == 'P')
                s.push(getID(set()));
            else if (cmd[0] == 'D')
                s.push(s.top());
            else
            {
                set x1 = getSet[s.top()];
                s.pop();
                set x2 = getSet[s.top()];
                s.pop();
                set x;
                if (cmd[0] == 'U')
                    set_union(x1.begin(), x1.end(), x2.begin(), x2.end(), inserter(x,x.begin()));
                else if (cmd[0] == 'I')
                    set_intersection(x1.begin(), x1.end(), x2.begin(), x2.end(), inserter(x, x.begin()));
                else if (cmd[0] == 'A')
                {
                    x = x2;
                    x.insert(getID(x1));
                }
                s.push(getID(x));
            }
            cout << getSet[s.top()].size() << endl;
        }
        cout << "***" << endl;
    }
    return 0;
}

你可能感兴趣的:(算法竞赛入门经典)