bzoj 2733: [HNOI2012]永无乡

最新推荐文章于 2022-09-07 19:41:55 发布

原创最新推荐文章于 2022-09-07 19:41:55 发布 · 329 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#treap #启发式

本文介绍了一种基于Treap的数据结构实现的启发式合并算法，该算法通过随机优先级平衡二叉搜索树来高效地完成合并操作。文章详细阐述了Treap的基本概念、旋转操作、插入节点以及合并过程，并提供了完整的C++实现代码。

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

treap启发式合并；

#include<bits/stdc++.h>
#define rep(i,k,n) for(int i=k;i<=n;i++)
#define ls ch[x][0]
#define rs ch[x][1]
#define son ch[x][d]
using namespace std;
const int N=100005;
int root[N],p[N],ch[N][2],s[N],val[N],key[N],n,m,tot=0,Q;
int find(int x){return p[x]==x ? x : p[x]=find(p[x]);}
void up(int x){s[x]=s[ls]+s[rs]+1;}
void rotate(int& x,int d){
    int y=ch[x][d^1];ch[x][d^1]=ch[y][d];ch[y][d]=x;up(x),x=y;
}
inline void ins(int&x,int k){
    if(!x){x=k;key[x]=rand();ls=rs=0;s[x]=1;return;}
    int d=val[k]>val[x];
    ins(ch[x][d],k);
    if(key[son]<key[x])rotate(x,d^1);
    up(x);
}
inline int kth(int x,int k){
    if(s[x]<k)return -1;
    int ss=s[ls]+1;
    if(k==ss)return x;
    if(k<ss)return kth(ls,k);
    else return kth(rs,k-ss);
}
void merge(int x,int& y){
    if(!x)return;
    merge(ls,y),merge(rs,y);
    ins(y,x);
}
void init(){
    rep(i,1,n){p[i]=i,ins(root[i],i);}
}
void link(int x,int y){
    x=find(x);y=find(y);
    if(x!=y){
        if(s[root[x]]>s[root[y]])swap(x,y);
        merge(root[x],root[y]);
        p[x]=y;
    }
}
int main(){//freopen("in.in","r",stdin);
    scanf("%d%d",&n,&m);int x,y;
    rep(i,1,n)scanf("%d",&val[i]);
    init();
    rep(i,1,m){
        scanf("%d%d",&x,&y);link(x,y);
    }scanf("%d",&Q);
    char ss[20];
    while(Q--){
        scanf("%s%d%d",ss,&x,&y);
        if(ss[0]=='B'){
            link(x,y);
            }
        else printf("%d\n",kth(root[find(x)],y));
    }
}