杭电多校:KD-Graph(并查集)

传送门

题意:

将图中的点划分为k个集合,求一个值d,对于集合内的点:若点与点之间有直接相连的路径,则这些路径至少有一个小于等于d;对于不同集合的点:若这两点有直接相连的路径,则这些路径全部大于d。

思路:

这题和点的关系不大,我们更看重边,因为要保证集合内的边的最大值小于等于某个数,集合与集合之间的边值大于某个数,因而首先一开始可以将每个点都看成一个单独的集合,此时,d的值为所有边的最大值+1,如果此时的集合数大于k,则需要将未合并的最小权值边的端点合并,然后再看合并完后集合数量是否等于k。
综上所述:可以得到一个行之有效的解法:将权值从小到大排列,以到达不同权值的边为分界点,每次判断当前集合的数目是否为k

#include<bits/stdc++.h>
using namespace std;
#define ll long long
struct node{
	int u,v;
	ll w;
}e[500010];
bool cmp(node a, node b)
{
	return a.w < b.w;
}
int f[1000010];
int find(int x)
{
	return f[x] == x ? x : f[x]=find(f[x]);
}

int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		int n,m,k;
		scanf("%d%d%d",&n,&m,&k);
		for(int i = 1; i <= n; i++)f[i] = i;
		for(int i = 1; i <= m; i++)
		{
			scanf("%d%d%lld",&e[i].u,&e[i].v,&e[i].w);
		}
		sort(e+1,e+1+m,cmp);
		ll ans = 0,cnt = 0;
		cnt = n;
		int flag = 0;
		for(int i = 1; i <= m; i++)
		{
			if(e[i].w != e[i-1].w)
			{
				if(cnt == k)
				{
					flag = 1;
					ans = e[i-1].w;
					break;
				}
			}
			if(find(e[i].u) == find(e[i].v))continue;
			f[find(e[i].u)] = find(e[i].v);
			cnt--;
			
		}
		if(cnt == k)
		{
			cout<<ans<<endl;
		}
		else
		{
			cout<<"-1"<<endl;
		}
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值