题目链接:QS Network - ZOJ 1586 - Virtual Judge (ppsucxtt.cn)
题目中可以切换至中文题意,我就不在这说明题意了。
这道题目与我们之前做的那种kruscal裸题并不完全一样,因为这道题目需要我们把点的权值也计算在内,而且不只是计算一次,我一开始的想法就是按照一般的kruscal来做,当确定要加入某条边时顺便把该边的两个端点的点权值也加一遍,然后就一直wa,我也没明白是为啥,仔细思考后发现这样并不一定是正确结果,因为我们只是按照边权值从小到大排序,这样并不能保证边权值加上两个点权值一定也是最小的,思考到这,就不难想到可以把点权值放到边权值中一块进行排序,这样就是一个裸的kruscal问题了,下面是代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
const int N=3e3+10;
int fu[N],fee[N],d[N][N];
struct node{
int u,v,w;
}p[N*N];
bool cmp(node a,node b)
{
return a.w<b.w;
}
int find(int x)
{
if(x!=fu[x]) fu[x]=find(fu[x]);
return fu[x];
}
int main()
{
int T,n;
cin>>T;
while(T--)
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&fee[i]);
fu[i]=i;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%d",&d[i][j]);
int cnt=0;
for(int i=1;i<=n;i++)
for(int j=i+1;j<=n;j++)
p[++cnt]={i,j,d[i][j]+fee[i]+fee[j]};//把点权值放进边权值
sort(p+1,p+cnt+1,cmp);
int count=0,ans=0;
for(int i=1;i<=cnt;i++)
{
int f1=find(p[i].u),f2=find(p[i].v);
if(f1==f2) continue ;
ans+=p[i].w;
fu[f1]=f2;
count++;
if(count==n-1) break;
}
printf("%d\n",ans);
}
return 0;
}