KD-Graph(并查集)

在这里插入图片描述
在这里插入图片描述
这道题难倒是不难,主要是题目的意思可能不太好理解,就是有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;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值