ZOJ 1586

最小生成树水题。。读好题就好了

每个人自己wifi 有个权值，链接还有个权值。。。直接克鲁斯卡尔。。。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct node{int u,v,cost;}a[1000005];
int pre[1000005];
inline int find(int x){return x==pre[x]?x:find(pre[x]);}
inline void unite(int x,int y){int t1=find(x),t2=find(y);pre[t1]=t2;}
inline bool cmp(node x,node y){return x.cost<y.cost;}
int cost[1000005];
int main()
{
    int n,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&cost[i]);
            pre[i]=i;
        }
        int cnt=0,num;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                scanf("%d",&num);
                a[cnt].u=i,a[cnt].v=j;
                a[cnt].cost=num+cost[i]+cost[j];
                cnt++;
            }
        }
        int tot=0,sum=0;
        sort(a,a+cnt,cmp);
        for(int i=0;i<cnt;i++)
        {
            if(find(a[i].u)!=find(a[i].v))
            {
                unite(a[i].u, a[i].v);
                sum += a[i].cost;
                tot ++;
            }
            if(tot == n-1) break;
        }
        printf("%d\n",sum);
    }
    return 0;
}