最小生成树+Kruskal算法并查集的使用

最新推荐文章于 2022-03-16 20:28:37 发布

H_ang

最新推荐文章于 2022-03-16 20:28:37 发布

阅读量266

点赞数

CC 4.0 BY-SA版权

分类专栏：最小生成树

本文链接：https://blog.youkuaiyun.com/qq_21433411/article/details/88134510

最小生成树专栏收录该内容

3 篇文章

订阅专栏

本文探讨了使用Kruskal算法解决最小生成树中寻找第K大边的问题，通过调整并查集实现，展示了完整的C++代码实现过程及调试经验。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：http://poj.org/problem?id=2349
题目大意：最小生成树找第k大边。

直接Kruskal就把最小生成树跑出来，找第k大边就ok。而我的模板并查集下标从1开始。但是我的点从0开始。导致WA了。
在这里插入图片描述

#include<bits/stdc++.h>
using namespace std;
#define LL long long

const int maxn=1e6+10;

struct Edge
{
    int l, r;
    double v;

}edge[maxn];

struct F
{
    double x;
    double y;

}f[maxn];

int cmp(Edge a, Edge b)
{
    return a.v<b.v;
}

int cmp1(Edge a, Edge b)
{
    return a.v>b.v;
}

int a[maxn];

int fd(int x)
{
    while(a[x]>=0)//从0开始：>=0 从1开始： >0
        x=a[x];
    return x;
}

void lj(int x, int y)
{
    x=fd(x);
    y=fd(y);

    if(x!=y)
    {
        if(a[x]<a[y])
            a[x]+=a[y],a[y]=x;
        else
            a[y]+=a[x],a[x]=y;
    }
}

vector<double> v;
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        v.clear();
        memset(a, -1, sizeof(a));
        int k, n, p=0;
        scanf("%d%d",&k,&n);
        for(int i=0;i<n;i++)
        {
            scanf("%lf%lf",&f[i].x,&f[i].y);
        }
        for(int i=0;i<n;i++)
        {
            for(int j=i+1;j<n;j++)
            {
                edge[p].v=sqrt((f[i].x-f[j].x)*(f[i].x-f[j].x)+(f[i].y-f[j].y)*(f[i].y-f[j].y));
                edge[p].l=i, edge[p].r=j, p++;
                edge[p].v=edge[p-1].v;
                edge[p].l=j, edge[p].r=i, p++;
            }
        }
        sort(edge, edge+p, cmp);
        for(int i=0;i<p;i++)
        {
            if(fd(edge[i].l)!=fd(edge[i].r))
            {
                v.push_back(edge[i].v);
                lj(edge[i].l, edge[i].r);
            }
        }
        sort(v.begin(), v.end(), greater<double>() );

        printf("%.2f\n",v[k-1]);

    }

    return 0;
}