hdu 3172 virtual friends 并查集

题意很简单就是告诉你一些话，然后就是说这两个人刚成为好朋友，然后问你就是到每句话的这两个人的朋友网里面有多少人。这个题目的输入比较麻烦啊，tle好几次。解法就是每读入一条边就用并查集的思想，修改祖先节点，然后修改包括祖先的朋友网的大小，然后输出祖先节点的朋友网的规模就可以了。

#include<iostream>
#include<stdio.h>
#include<string>
#include<map>
using namespace std;
map<string,int> A;
int father[100005],sum[100005];
int find_ant(int n)
{
	if(n!=father[n])
		father[n]=find_ant(father[n]);
	return father[n];
}
void unin(int x,int y)
{
	x=find_ant(x);
	y=find_ant(y);
	if(x!=y)
	{
		father[x]=y;
		sum[y]=sum[y]+sum[x];
	}
}
int main()
{
	int t,n,total;
	string a,b;
	while(scanf("%d",&t)!=EOF)
	{
		while(t--)
		{
			cin>>n;
			total=0;
			A.clear();
			while(n--)
			{
				cin>>a>>b;
				if(A.find(a)==A.end())
				{
					total=total+1;
					A[a]=total;
					father[total]=total;
					sum[total]=1;
				}
				if(A.find(b)==A.end())
				{
					total=total+1;
					A[b]=total;
					father[total]=total;
					sum[total]=1;
				}
				unin(A[a],A[b]);
				int ans=find_ant(A[b]);
				printf("%d\n",sum[ans]);
			}
		}
	}
	return 0;
}