首页 > 代码库 > uva 12096 - The SetStack Computer(STL)

uva 12096 - The SetStack Computer(STL)

UVA 12096 - The SetStack Computer

题目链接

题意:几个操作,push是在栈顶加入一个空集,dup是复制栈顶集合,在放入栈顶,union是把头两个取并集放回,int是头两个取交集放回,add是取头两个,把第一个当成一个集合加入第二个,每次操作输出栈顶集合的里面的个数

思路:用set,stack模拟,然后利用map去hash一个集合,模拟即可

代码:

#include <cstdio>
#include <cstring>
#include <stack>
#include <algorithm>
#include <set>
#include <map>
using namespace std;

int t, n, hn;
char op[10];
stack<set<int> > st;
map<set<int>, int> hash;
set<int> s1, s2;
set<int>::iterator it;

void print() {
    printf("%d\n", (int)st.top().size());
}

void ad(set<int> s) {
    if (hash.count(s))
	return;
    hash[s] = hn++;
}

void pus() {
    set<int> s;
    ad(s);
    st.push(s);
    print();
}

void dup() {
    set<int> s = st.top();
    st.push(s);
    print();
}

void uni() {
    s1 = st.top(); st.pop();
    s2 = st.top(); st.pop();
    for (it = s1.begin(); it != s1.end(); it++)
	s2.insert(*it);
    ad(s2);
    st.push(s2);
    print();
}

void ins() {
    s1 = st.top(); st.pop();
    s2 = st.top(); st.pop();
    set<int> s3;
    it = s1.begin();
    set<int>::iterator it2 = s2.begin();
    while (it != s1.end() && it2 != s2.end()) {
	if (*it < *it2) *it++;
	else if (*it > *it2) it2++;
	else {
	    s3.insert(*it);
	    it++;
	    it2++;
	}
    }
    ad(s3);
    st.push(s3);
    print();
}

void add() {
    s1 = st.top(); st.pop();
    s2 = st.top(); st.pop();
    s2.insert(hash[s1]);
    ad(s2);
    st.push(s2);
    print();
}

int main() {
    scanf("%d", &t);
    while (t--) {
	hn = 0;
	hash.clear();
	while (!st.empty()) st.pop();
	scanf("%d", &n);
	while (n--) {
	    scanf("%s", op);
	    if (op[0] == 'P')
		pus();
	    if (op[0] == 'D')
		dup();
	    if (op[0] == 'A')
		add();
	    if (op[0] == 'U')
		uni();
	    if (op[0] == 'I')
		ins();
	}
	printf("***\n");
    }
    return 0;
}