nyoj38(布线问题)(kruskal算法)

来源:互联网 发布:alibaba json jar 编辑:程序博客网 时间:2024/05/22 12:20
#include<stdio.h>
#include<algorithm>
using namespace std;
int father[520];
int length[520];
struct edge
{
    int x,y,value;
}a[130000];
bool cmp(edge a,edge b)
{
    return a.value<b.value;
}
void set(int n)
{
    for(int i=1;i<=n;i++)
    father[i]=i;
}
int find(int x)
{
    if(x==father[x])
    return father[x];
    return find(father[x]);
}
void merge(int a,int b)
{
    int x=find(a);
    int y=find(b);
    if(x<=y)
    father[y]=x;
    else
    father[x]=y;
}
int kruskal(int m,int n)
{
    set(m);
    for(int i=0;i<n;i++)
    scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].value);
    sort(a,a+n,cmp);
    for(int i=0;i<m;i++)
    scanf("%d",&length[i]);
    sort(length,length+m);
    int sum=0;
    for(int i=0;i<n;i++)
    {
        if(find(a[i].x)!=find(a[i].y))
        {
            sum+=a[i].value;
            merge(a[i].x,a[i].y);
        }
    }
    return sum+length[0];
}
int main()
{
    int icase;
    scanf("%d",&icase);
    while(icase--)
    {
        int v,e;
        scanf("%d%d",&v,&e);
        printf("%d\n",kruskal(v,e));
    }
    return 0;
}

















0 0
原创粉丝点击