zoj 1586 QS Network

#include<bits/stdc++.h>
using namespace std;
int edge[1024][1024];

int main()
{
    int ans,_,i,j,k,n,w,a[1024],lowcost[1024];
    scanf("%d",&_);
    while(_--)
    {
        scanf("%d",&n);
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
        for(i=1;i<=n;i++)
            for(j=1;j<=n;j++)
        {
            scanf("%d",&w);
            if(i==j) edge[i][j]=1024*10;
            else
            {
                edge[j][i]=edge[i][j]=w+a[i]+a[j];
            }
        }

        ans=0;
        for(i=1;i<=n;i++)
        {
            lowcost[i]=edge[1][i];
        }
        lowcost[1]=-1;
        for(i=2;i<=n;i++)
        {
            int t=1024*10;
            for(k=1;k<=n;k++)
            {
                if(lowcost[k]!=-1&&lowcost[k]<t)
                {
                    j=k;
                    t=lowcost[k];
                }
            }
            ans+=t;
            lowcost[j]=-1;
            for(k=1;k<=n;k++)
            {
              lowcost[k]=min(lowcost[k],edge[j][k]);
            }
        }
        printf("%d
",ans);
    }
    return 0;
}