网络交友（并查集问题）

题目描述

在网络社交的过程中，通过朋友，也能认识新的朋友。在某个朋友关系图中，假定 A 和 B 是朋友，B 和 C 是朋友，那么 A 和 C 也会成为朋友。即，我们规定朋友的朋友也是朋友。
现在要求你每当有一对新的朋友认识的时候，你需要计算两人的朋友圈合并以后的大小。

输入描述

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

输出描述

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

样例输入

3
Fred Barney
Barney Betty
Betty Wilma

样例输出

2
3
4

AC代码

#include<iostream>
#include<map>
using namespace std;
int father[5005];//父亲结点，结点数；
int size[5005];//合并以后朋友圈的大小
map<string ,int> m;
void init(int n){
for(int i = 0;i<5005;i++){
father[i] = i,size[i] = 1;
}
}
int get(int x){
   if(father[x]==x) {
       return x;
   }
   return father[x] = get(father[x]);
}
void merge(int a,int b){
    a = get(a);
    b = get(b);
    if(a==b){
        cout<<size[b]<<endl;
        return;
    }
        father[b] = a;
        size[a]+=size[b];
        cout<<size[a]<<endl;

}
int main(){
    int n;
    string a1,a2;
    cin>>n;
    int num = 1;
    init(n);
    for(int i = 0;i<n;i++){
        cin>>a1>>a2;
        if(!m.count(a1)){
            m[a1] = num;
            num++;
        }
        if(!m.count(a2)){
            m[a2] = num;
            num++;
        }
        merge(m[a1],m[a2]);
    }
    return 0;
}