zoj 1586 最小生成树

点击打开链接
#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxv=1000+5;
const int maxe=500000+5;
struct Edge{
int from,to,weight;
};
bool operator<(const Edge &E1,const Edge &E2)
{
    return E1.weight<E2.weight;
}
Edge edges[maxe];
int cost[maxv];
int parent[maxv];
int find(int x)
{
    return x==parent[x]?x:parent[x]=find(parent[x]);
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(cost,0,sizeof(cost));
        int n;      //QS数量
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            {
                scanf("%d",&cost[i]);
                parent[i]=i;
            }
        int cnt=0;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
        {
            int k;
            scanf("%d",&k);
            if(j>i) {
                edges[cnt].from=i;
                edges[cnt].to=j;
                edges[cnt].weight=k+cost[i]+cost[j];                     //此处有坑点：必须提前加，否则会影响kruskal算法对边的选择
                cnt++;
            }
        }
        sort(edges,edges+cnt);
        int sum=0;
        for(int i=0;i<cnt;i++)
        {
            int p1=find(edges[i].from);
            int p2=find(edges[i].to);
            if(p1==p2) continue;
            parent[p1]=p2;
            sum+=edges[i].weight;
        }
        printf("%d\n",sum);
    }
    return 0;
}