ZOJ 3641 <并查集+STL>

题意：

给出order数n

给出n行order<3种>

第一种：travel A m a1 .. ai .. an表示来了同学A 其中知道m个信息 信息号为a1 .. ai .. an..

第二种：share A B表示 A和B同学分享了她们共有的信息..其中信息可传递..并且和share先后无关..

第三种：check A 表示问A同学知道了多少条信息

思路：

用map<string, int> 来让同学名字变成序号..

用set来对信息去重..

用并查集使资源共享~<可节省内存~>

用set里面的size可以求的根节点的大小~即share后掌握的知识点..

Tips：

set里完成节点之间的并查后都要给子节点clear()..然后下次再找根节点的大小就好了~

Code：

 

#include <stdio.h>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <map>
#include <set>
using namespace std;
const int MAXN = 500010;

int f[MAXN];
set<int> s[MAXN];

int find(int x)
{
    return f[x]==x?x:f[x] = find(f[x]);
}

int main()
{
    int i, j, k;
    int n;
    string order, name, name2;
    int m, tmp, tot;
    while(cin >> n)
    {
        for(i = 0; i <= n; ++i){
            f[i] = i;
            s[i].clear();
        }
        map<string, int> mm;
        tot = 0;

        for(k = 0; k < n; ++k){
            cin >> order;
            if(order == "arrive"){
                cin >> name;
                if(mm[name] == 0){
                    mm[name] = ++tot;
                }

                cin >> m;
                while(m--)
                {
                    cin >> tmp;
                    s[mm[name]].insert(tmp);
                }
            }
            else if(order == "share"){
                cin >> name >> name2;
                int x = mm[name], y = mm[name2];
                int xx = find(x), yy = find(y);
                if(xx != yy){
                    f[xx] = yy;
                    set<int>::iterator it;
                    for(it = s[xx].begin(); it != s[xx].end(); ++it)
                    s[yy].insert(*it);
                    s[xx].clear();
                }
            }
            else if(order == "check"){
                cin >> name;
                int x = find(mm[name]);
                cout << s[x].size() << endl;
            }
        }
    }

    return 0;
}