UVA12096 - The SetStack Computer(set + map映射)

UVA12096 - The SetStack Computer(set + map映射)

题目链接

题目大意：有五个动作：

push ： 把一个空集合{}放到栈顶。

dup ： 把栈顶的集合取出来，在入栈两次。

add ： 出栈两次，把第一个集合作为一个元素放入第二个集合中，再将第二个集合入栈

union： 出栈两次，取这两个集合的并集，将结果入栈。

intersect： 出栈两次，取这两个集合的交集，将结果入栈。

每次执行动作后还需要输出目前栈顶集合的元素个数。

解题思路：这题可以用栈和set来模拟，push就把空的集合入栈，但是在并集和交集的时候就需要判段集合是否相同，所以这里可以用map把出现过的集合手动的映射成数字。

代码：

#include <cstdio>
#include <cstring>
#include <stack>
#include <set>
#include <map>

using namespace std;

char op[15];
int n;

typedef set<int> E;
stack<E> s;
map<E, int> vis;
E tmp1, tmp2;
set<int>::iterator it;
int num;

void hash (E a) {

if (!vis.count(a))
vis[a] = ++num;
}

void Push () {

tmp1.clear();
s.push (tmp1);
}

void Dup () {

tmp1 = s.top();
s.push (tmp1);
}

void Add () {

tmp2 = s.top();
s.pop();
tmp1 = s.top();
s.pop();
tmp1.insert (vis[tmp2]);
hash(tmp1);
s.push(tmp1);
}

void Union () {

tmp2 = s.top();
s.pop();
tmp1 = s.top();
s.pop();
for (it = tmp1.begin(); it != tmp1.end(); it++)
tmp2.insert (*it);
hash (tmp2);
s.push (tmp2);
}

void Intersect () {

tmp2 = s.top();
s.pop();
tmp1 = s.top();
s.pop();
E tmp;
for (it = tmp1.begin(); it != tmp1.end(); it++)
if (tmp2.count(*it))
tmp.insert (*it);
hash (tmp);
s.push(tmp);
}

void solve () {

switch (op[0]) {

case 'P' : Push();     break;
case 'D' : Dup();      break;
case 'U' : Union();    break;
case 'I' : Intersect(); break;
case 'A' : Add();      break;
}
printf ("%d\n", s.top().size());
}

void init () {

num = 1;
while (!s.empty()) {
s.pop();
}
vis.clear();
}

int main () {

int T;
scanf ("%d", &T);
while (T--) {

scanf ("%d", &n);
while (n--) {
scanf ("%s", op);
solve();
}
printf ("***\n");
}
return 0;
}