ZOJ3641 Information Sharing

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4809

题意：

给出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> 来让同学名字变成序号..

share则使用并查集，对人名使用并查集。share时合并则使用set的插入，优化：让size小的插入size大的。

#include <cstdio>#include <cstring>#include <cmath>#include <map>#include <set>#include <vector>#include <iostream>#include <algorithm>#include<string>using namespace std;//const double eps=1e-7;//const double INF=1e50;//const double pi=acos(-1);#define N 100005#define M 1000005int f[N],n,people;char order[10],name1[20],ch,name2[20];string na,nb;map<string,int> a;map<int,int> info;set<int> b[N];int root(int s){    if (f[s]==s) return s;    f[s]=root(f[s]);    return f[s];    /*int temp=s;    while(f[temp]!=temp)temp=f[temp];    while(f[s]!=s){s=f[s];f[s]=temp;}    return temp;*/}int main(){    //freopen("a","r",stdin);    int i,m,j,k,in;    while (scanf("%d",&n)!=EOF)    {        a.clear();        for (i=0;i<n;i++) b[i].clear();        info.clear();        people=0;        in=0;        for (i=1;i<=n;i++)        {            gets(name1);            scanf("%s %s",order,name1);            na=name1;            if (order[0]=='a')            {                a[na]=people;                f[people]=people;                scanf(" %d",&m);                map<int,int>::iterator it;                int ss;                for (j=1;j<=m;j++)                {                    scanf("%d",&k);                    it=info.find(k);                    if (it==info.end())                    {                        in++;                        info[k]=in;//不要写反了                        ss=in;                    }                    else ss=(*it).second;                    b[people].insert(ss);                }                people+=1;            }            else if (order[0]=='s')            {                int n1,n2,n3;                scanf(" %s",name2);                nb=name2;                map<string,int>::iterator it;                it=a.find(na);                n1=(*it).second;                it=a.find(nb);                n2=(*it).second;                n1=root(n1);//找根不可缺少                n2=root(n2);                if (n1==n2) continue;                if (b[n1].size()<b[n2].size())                {                    n3=n1;                    n1=n2;                    n2=n3;                }                set<int>::iterator it1;                for (it1=b[n2].begin();it1!=b[n2].end();it1++) b[n1].insert(*it1);                b[n2].clear();                f[n2]=n1;            }            else            {                int n1;                map<string,int>::iterator it;                it=a.find(na);                n1=(*it).second;                n1=root(n1);                printf("%d\n",b[n1].size());            }        }    }    return 0;}