在网络社交的过程中,通过朋友,也能认识新的朋友。在某个朋友关系图中,假定 A 和 B 是朋友,B 和 C 是朋友,那么 A 和 C 也会成为朋友。即,我们规定朋友的朋友也是朋友。
现在要求你每当有一对新的朋友认识的时候,你需要计算两人的朋友圈合并以后的大小。
输入格式
第一行:一个整数 n(n\leq 5000)n(n≤5000),表示有 nn 对朋友认识。
接下来 nn 行:每行输入两个名字。表示新认识的两人的名字,用空格隔开。(名字是一个首字母大写后面全是小写字母且长度不超过 20 的串)。
输出格式
对于每一对新认识的朋友,输出合并以后的朋友圈的大小。
#include <iostream>
#include <cstring>
#include <map>
using namespace std;
const int N=5e3+9;
//int n;
int fa[N],size[N];
map<string,int> friendship;
int cnt=0;
void init(){
for(int i=0;i<N;i++){
fa[i]=i;
size[i]=1;
}
}
int get(int x){
if(fa[x]==x){
return x;
}else{
return fa[x]=get(fa[x]);
}
}
void merge(int x,int y){
int dx=get(x);
int dy=get(y);
if(dx!=dy){
fa[dx]=dy;
size[dy]+=size[dx];
}
}
int main(){
init();
int n;
cin>>n;
while(n--){
string a,b;
cin>>a>>b;
if(friendship.find(a)==friendship.end()){
friendship[a]=cnt++;
}
if(friendship.find(b)==friendship.end()){
friendship[b]=cnt++;
}
int x=friendship[a];
int y=friendship[b];
merge(x,y);
cout<<size[get(x)]<<endl;
}
return 0;
}