字典树+并查集Virtual Friends HDU 3172（900MS）

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
int f[100010];
int sum[100010];
int c;
int tot;
int find(int i)                               //并查集查找父项
{
while(f[i]!=i)
i=f[i];
return f[i];
}
struct trie
{
trie *next[60];
int num;
bool v;
}*q;
trie memory[100010];
int build(char *str)                                    //建立字典树
{
int len=strlen(str);
trie *p=&memory[0];
int flag=0;
for(int i=0;i<len;i++)
{
int id=str[i]-'A';
if(p->next[id]==NULL)
{
p->v=false;
p->next[id]=&memory[++c];
flag=1;
}
p=p->next[id];
}
if(p->v==false)                         //以防新的名字为已有名字的前缀
{
flag=1;
p->v=true;
}
tot+=flag;                                    //统计元素总数
if(flag==0)
return p->num;                //返回该名字的序号
else
{
p->num=tot;
return p->num;
}
}
int main()
{
int n;
int t,i;
int max;
int t1,t2;
char name1[25],name2[25];
while(scanf("%d",&t)!=-1)
{
while(t--)
{
memset(memory,0,sizeof(memory));
tot=c=0;
for(i=1;i<100005;i++)
{
f[i]=i;
sum[i]=1;
}
scanf("%d",&n);
max=0;
for(i=0;i<n;i++)
{
scanf("%s%s",&name1,&name2);
t1=build(name1);
t2=build(name2);
t1=find(t1);
t2=find(t2);
if(t1!=t2)
{
f[t1]=t2;
sum[t2]+=sum[t1];                   //直接将该项的子项数相加，省去在搜索并查集，减少复杂度和时间
}
printf("%d\n",sum[t2]);
}
}
}
return 0;
}