ZOJ 1586 QS Network

//没啥好说的,一道水题而已,直接看代码:

AC代码:

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 1005
int p[N];
int a[N];
int n;
int L;
struct node
{
    int w,v,c;
}t[N*N];
int cmp(node p,node q)
{
    return p.c<q.c;
}
int find(int k)
{
    if(p[k]==k)
    {
        return p[k];
    }
    else
    {
        return find(p[k]);
    }
}
int cl()
{
    int sum=0,cnt=0;
    int i;
    for(i=0;i<L;i++)
    {
        int x=find(t[i].w);
        int y=find(t[i].v);
        if(x!=y)
        {
            p[y]=x;
            cnt++;
            sum+=t[i].c;
        }
        if(cnt==n-1)
            return sum;
    }
    return sum;
}
int main()
{
    int e;
    scanf("%d",&e);
    while(e--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            p[i]=i;
        }
        int k;
        L=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                scanf("%d",&k);
                t[L].w=i;
                t[L].v=j;
                t[L].c=k+a[i]+a[j];
                L++;
            }
        }
        sort(t,t+L,cmp);
        printf("%d\n",cl());
    }
    return 0;
}

你可能感兴趣的:(ZOJ 1586 QS Network)