蓝桥杯：网络交友

在网络社交的过程中，通过朋友，也能认识新的朋友。在某个朋友关系图中，假定 A 和 B 是朋友，B 和 C 是朋友，那么 A 和 C 也会成为朋友。即，我们规定朋友的朋友也是朋友。

现在要求你每当有一对新的朋友认识的时候，你需要计算两人的朋友圈合并以后的大小。

输入格式

第一行：一个整数 n(n≤5000)，表示有n对朋友认识。

接下来n行：每行输入两个名字。表示新认识的两人的名字，用空格隔开。（名字是一个首字母大写后面全是小写字母且长度不超过 20 的串）。

输出格式

对于每一对新认识的朋友，输出合并以后的朋友圈的大小。

样例输入

3

Fred Barney

Barney Betty

Betty Wilma

样例输出

2

3

4

#include <iostream>
#include<map>
using namespace std;

int father[10001],size[10001];
map<string,int>m;

int get(int x){
	if(father[x]==x) return x;
	
	int y=father[x];
	father[x]=get(y);
	
	return father[x];
}
void merge(int a,int b){
	a=get(a);
	b=get(b);
	if(a!=b){/*两个集合合并后size为两个集合size相加*/
        int sum=size[a]+size[b];
        size[a]=size[b]=sum;
		father[a]=b;
	}
}

int main(){
	int n,count=1;
	/*初始化*/
	for(int i=1;i<=10000;i++){
		father[i]=i;
		size[i]=1;
	}
	cin>>n;
	/*输入两个人名，用map给他们编号，将他们的编号合并为一个集合*/
	while(n--){
		string s1,s2;
		cin>>s1>>s2;
		if(m.count(s1)<1) m[s1]=count++;
		if(m.count(s2)<1) m[s2]=count++; 
		merge(m[s1],m[s2]);
		cout<<size[get(m[s1])]<<endl;
	}
	
	

	return 0;
}