【并查集】-HDU-3172-Virtual Freinds

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3172点击打开链接

题目描述：

输入N个人的关系，输出你最新输入的两个人所在社交分组的人数（相互有朋友关系的视为一个组）。

解题思路：

另设一个RANK数组来存储对应集合的元素个数，在merge合并操作的同时将数组合并即得到答案。对字符串处理用MAP，不然会超时，我就T了N多遍，后来干脆去现学现卖了MAP。

AC代码：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <map>

using namespace std;

map <string,int>my_map;
int arr[100050],mem[100050];
int q;

void reset()
{
    my_map.clear();
    q=1;
    memset(arr,0,sizeof(arr));
    int i;
    for(i=0;i<100050;i++)
    {
        arr[i]=i;
        mem[i]=1;
    }
}

int find(int x)
{
    return x==arr[x]?x:arr[x]=find(arr[x]);
}

void mer(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy&&mem[fx]<=mem[fy])
    {
        mem[fy]=mem[fx]+mem[fy];
        arr[fx]=fy;
    }
    else if(fx!=fy&&mem[fx]>mem[fy])
    {
        mem[fx]=mem[fx]+mem[fy];
        arr[fy]=fx;
    }
}

int main()
{
    reset();
    int T,x,y,i;
    char a[22],b[22];
    while(scanf("%d",&T)!=EOF)
        while(T--)
        {
            int n;
            scanf("%d",&n);
            while(n--)
            {
                scanf("%s%s",a,b);
                if(!my_map[a])my_map[a]=q++;
                if(!my_map[b])my_map[b]=q++;
                x=my_map[a];
                y=my_map[b];
                mer(x,y);
                printf("%d\n",mem[find(x)]);
            }
            reset();
        }
    return 0;
}

AC截图：