uva 12096 - The SetStack Computer(STL)

標籤：style   http   color   os   io   for   ar   代碼   

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;}