(ZOJ - 1586)QS Network(kruscal)

题目链接: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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值