ZOJ 3641 Information Sharing

今天老是细节上出错，很不爽。

并查集，维护set数组，注意及时释放内存。STL果然不是一般的大常数……

#include <algorithm>
#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>

#include <map>
#include <set>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <vector>
#include <string>
#include <bitset>
#include <memory>
#include <complex>
#include <numeric>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <time.h>
#include <ctype.h>
#include <locale.h>

using namespace std;

#pragma pack(4)

const double  eps = 1e-8;
const double   pi = acos(-1.0);
const int     inf = 0x7f7f7f7f;

#define   at(a,i) ((a)&(1<<(i)))
#define   nt(a,i) ((a)^(1<<(i)))
#define set1(a,i) ((a)|(1<<(i)))
#define set0(a,i) ((a)&(~(1<<(i))))

#define gret(a,b) (((a)-(b))>eps)
#define less(a,b) (((b)-(a))>eps)
#define greq(a,b) (((a)-(b))>-eps)
#define leeq(a,b) (((b)-(a))>-eps)
#define equl(a,b) (fabs((a)-(b))<eps)

#define lmax(a,b) ((a)>(b)?(a):(b))
#define lmin(a,b) ((a)<(b)?(a):(b))
#define fmax(a,b) (gret(a,b)?(a):(b))
#define fmin(a,b) (less(a,b)?(a):(b))

const int MAXV = 100001;

int pa[MAXV];
set<int> info[MAXV];

int getPa(int i)
{
    if(pa[i]!=i) return pa[i]=getPa(pa[i]);
    else return i;
}
void Union(int i,int j)
{
    int x=getPa(i);
    int y=getPa(j);
    if(x!=y)
    {
        info[pa[x]=y].insert(info[x].begin(),info[x].end());
        info[x].clear();
    }
}

int n,m,w;
string o,u,v;
map<string,int> name;

int main()
{
    #ifndef ONLINE_JUDGE
    freopen("Information Sharing.txt","r",stdin);
    #endif

    while(cin>>n)
    {
        for(int i=1;n>=i;i++)
        {
            pa[i]=i;
            info[i].clear();
        }

        int cnt=0;
        name.clear();

        for(int i=1;n>=i;i++)
        {
            cin>>o;
            if(o=="arrive")
            {
                cin>>u>>m;
                if(!name[u]) name[u]=++cnt;
                while(m--)
                {
                    cin>>w;
                    info[getPa(name[u])].insert(w);
                }
            }
            else if(o=="share")
            {
                cin>>u>>v;
                if(!name[u]) name[u]=++cnt;
                if(!name[v]) name[v]=++cnt;
                Union(name[u],name[v]);
            }
            else if(o=="check")
            {
                cin>>u;
                if(!name[u]) name[u]=++cnt;
                cout<<info[getPa(name[u])].size()<<endl;
            }
        }
    }
    return 0;
}