Kruskal算法

题目信息
运行结果
本题排行
讨论区

布线问题
时间限制：1000 ms | 内存限制：65535 KB
难度：4

描述
南阳理工学院要进行用电线路改造，现在校长要求设计师设计出一种布线方式，该布线方式需要满足以下条件：
1、把所有的楼都供上电。
2、所用电线花费最少

输入
第一行是一个整数n表示有n组测试数据。(n<5)
每组测试数据的第一行是两个整数v,e.
v表示学校里楼的总个数(v<=500)
随后的e行里，每行有三个整数a,b,c表示a与b之间如果建铺设线路花费为c(c<=100)。（哪两栋楼间如果没有指明花费，则表示这两栋楼直接连通需要费用太大或者不可能连通）
随后的1行里，有v个整数,其中第i个数表示从第i号楼接线到外界供电设施所需要的费用。( 0

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
int p[505];
int find(int x){
    return p[x]==x?x:p[x]=find(p[x]);
}
struct node{
    int x;
    int y;
    int w;
};
int cmp(node a,node b){
    return a.w<b.w;
}
int main(){
    int k;
    scanf("%d",&k);
    while(k--){
        int v,e,sum=0;
        scanf("%d%d",&v,&e);
        for(int i=1;i<=v;i++)
            p[i]=i;
        node map[e+5];
        for(int i=0;i<e;i++){
            scanf("%d%d%d",&map[i].x,&map[i].y,&map[i].w);
        }
        int minn=-1,ui;
        for(int i=0;i<v;i++){
            scanf("%d",&ui);
            if(minn==-1)
                minn=ui;
            else
                if(minn>ui)
                minn=ui;
        }
        sort(map,map+e,cmp);
        /*for(int i=0;i<e;i++)
            printf("%d ",map[i].w);*/
        for(int i=0;i<e;i++){
            int tt=find(map[i].x);
            int yy=find(map[i].y);
            if(tt!=yy){
                p[tt]=yy;
                sum+=map[i].w;
            }
        }
        printf("%d\n",sum+minn);
    }
    return 0;
}