poj 2985 并查集+树状数组求第k大数

liuwq012

于 2015-11-05 16:53:13 发布

阅读量198

点赞数

CC 4.0 BY-SA版权

分类专栏： acm 文章标签： acm 树状数组算法

本文链接：https://blog.youkuaiyun.com/liuwq012/article/details/84748475

acm 专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种结合并查集与线段树的数据结构实现，用于解决区间合并问题，并通过一个具体的C++代码示例展示了如何更新区间元素及查找特定排名的元素。

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


#include <stdio.h>
#include <string.h>
#define lowbit(i) i&(-i)
#define maxn 300000
using namespace std;
int a[maxn],c[maxn],p[maxn];
int fd(int x)
{
    return x==p[x] ? x :fd(p[x]);
}
void update(int x,int val)
{
    while(x<=maxn)
    {
        c[x]+=val;
        x+=lowbit(x);
    }
}
int find_kth(int k)
{
    int ans = 0,cur = 0;
    for(int i=20;i>=0;i--)
    {
        ans+=(1<<i);
        if(ans>maxn||cur+c[ans]>=k)
            ans-=(1<<i);
        else
            cur+=c[ans];
    }
    return ans+1;
}
int main()
{
    int n,m,k,l,t,q,x,y;
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        p[i]=i;
        a[i]=1;
    }
    update(1,n);
    int num = n;
    while(m--)
    {
        scanf("%d",&q);
        if(q==0)
        {
            scanf("%d %d",&x,&y);
            x = fd(x);
            y = fd(y);
            if(x==y) continue;
            update(a[x],-1);
            update(a[y],-1);
            update(a[x]+a[y],1);
            p[y]=x;
            a[x]+=a[y];
            num--;
        }
        else
        {
            scanf("%d",&k);
            printf("%d\n",find_kth(num-k+1));
        }
    }
    return 0;
}