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\n",ans);
}
return 0;
}