这道题难倒是不难,主要是题目的意思可能不太好理解,就是有n个点,m条边,分成k组,每组至少有一个点,这时我们有一个值d,保证在同一个组内的点之间的边都要小于等于这个d,不在同一个组的点之间的边都要大于这个d,我们可以通过并查集的方法对这个d值进行一个查找:
我们先对所有的边进行一个从小到大的排序,然后每次处理边相同的点进行集合合并,如果这一组(边相同的若干条边)处理为之后分成的集合数大于K,那么继续进行,如果等于K,说明KD图存在,如果小于K,说明KD图不存在,输出-1即可。
完整代码如下:
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1000000 + 10;
struct infor{
int a,b,cost;
}e[maxn];
int f[maxn];
bool cmp(infor x,infor y){
return x.cost < y.cost;
}
int find(int a){
while(a!=f[a]){
a = f[a] = f[f[a]];
}
return a;
}
int main(){
int t;
cin >> t;
while(t--){
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
for (int i=0;i<m;i++) scanf("%d%d%d",&e[i].a,&e[i].b,&e[i].cost);
if(n == k) cout << 0 << endl;
else{
int ans;
for(int i=1;i<=n;i++) f[i] = i;
sort(e,e+m,cmp);
int cnt = 0;
int len = e[0].cost;
while(cnt < m && n>k){
ans = e[cnt].cost;
while(cnt < m && e[cnt].cost == len){
int a = e[cnt].a;
int b = e[cnt].b;
a = find(a);
b = find(b);
if(a!=b){
f[a] = b;
n--;
}
cnt++;
}
if(cnt < m) len = e[cnt].cost;
else break;
}
if(n==k){
cout << ans << endl;
}
else cout << -1 << endl;
}
}
return 0;
}