NYOJ 38 布线问题 （并查集）

题目：NYOJ 38 布线问题

自己的一点思路，耗时比较大

//AC
#include <stdio.h>
#include <string.h>
#define INF 0x3f3f3f3f
#include <algorithm>
using namespace std;
struct node{
    int a;
    int b;
    int c;
}num[125010]; 

int pre[505];

int cmp(node x,node y)
{
    if(x.c<y.c)
        return 1;
    else
        return 0;
}

int find(int x)
{
    int r=x;
    while(pre[r]!=r)
        r=pre[r];
    int i=x,j;
    while(pre[i]!=r)
    {
        j=pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}

void merge(int x,int y)
{
    int fx=find(x),fy=find(y);
    if(fx!=fy)
    {
        pre[fy]=fx;
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int v,e,i;
        scanf("%d %d",&v,&e);
        for(i=1;i<=e;i++)
            scanf("%d %d %d",&num[i].a,&num[i].b,&num[i].c);
        sort(num+1,num+e+1,cmp);

        int ans=INF,far;
        for(i=0;i<v;i++)      //连接外界的最小花费 
        {
            scanf("%d",&far);
            if(far<ans)
                ans=far;
        }

        for(i=1;i<=v;i++)
            pre[i]=i; 

        for(i=1;i<=e;i++)     //构建最小生成树 
        {
            if(find(num[i].a)==find(num[i].b))
                continue;
            merge(num[i].a,num[i].b);
            ans+=num[i].c;
        }
        printf("%d\n",ans);
    }
    return 0;
}