首页 > 代码库 > HDU 3172 Virtual Friends(带权并查集)

HDU 3172 Virtual Friends(带权并查集)

题目地址:HDU 3172

带权并查集水题。每次合并的时候维护一下权值。注意坑爹的输入。。

代码如下:

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>

using namespace std;
#define LL long long
int bin[210000], num[210000];
int find1(int x)
{
    return bin[x]==x?x:bin[x]=find1(bin[x]);
}
void join(int x, int y)
{
    int f1=find1(x);
    int f2=find1(y);
    if(f1!=f2)
    {
        bin[f2]=f1;
        num[f1]+=num[f2];
        printf("%d\n",num[f1]);
    }
    else
        printf("%d\n",num[f2]);
}
int main()
{
    int t, n, i, cnt;
    char s1[30], s2[30];
    map<string,int>q;
    while(scanf("%d",&t)!=EOF)
    {
        while(t--)
        {
            scanf("%d",&n);
            for(i=1; i<210000; i++)
            {
                bin[i]=i;
                num[i]=1;
            }
            q.clear();
            cnt=1;
            while(n--)
            {
                getchar();
                scanf("%s%s",s1,s2);
                if(!q[s1])
                    q[s1]=cnt++;
                if(!q[s2])
                    q[s2]=cnt++;
                join(q[s1],q[s2]);
            }
        }
    }
    return 0;
}


HDU 3172 Virtual Friends(带权并查集)