UVA ~ 12096 ~ The SetStack Computer (模拟,statck + map + set + vector)

题意: 有一个专门为了集合运算而设计的“集合栈”计算机。该机器有一个开始为空的栈并且支持以下操作。
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;
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值