题意:
将图中的点划分为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;
}
}
}