【BZOJ2733】永无乡

题目链接:传送门
题解:
splay启发式合并,每次暴力拆较小的子树插入到另一个子树中

//by sdfzchy
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ls t[x].ch[0]
#define rs t[x].ch[1]
#define pa t[x].fa
using namespace std;
typedef long long LL;
const int inf=(1<<30),N=100010,mod=1e9+7;
int n,m,fa[N],val[N];
inline int in()
{
    char tmp=getchar();
    int res=0,f=1;
    while((tmp<'0'||tmp>'9')&&tmp!='-')tmp=getchar();
    if(tmp=='-') f=-1,tmp=getchar();
    while(tmp>='0'&&tmp<='9')   res=(res<<1)+(res<<3)+(tmp^48),tmp=getchar();
    return res*f;
}
inline int find(int x) {return fa[x]==x?fa[x]:fa[x]=find(fa[x]);}
struct Splay
{
    int cnt=0,root=0;
    struct node
    {
        int ch[2],fa,siz,val;
    }t[N];
    void init() {for(int i=1;i<=n;i++) t[i].val=val[i],t[i].siz=1;}
    inline int gi(int x) {return t[pa].ch[1]==x;}
    inline void upd(int x) {t[x].siz=t[ls].siz+t[rs].siz+1;}
    void rot(int x)
    {
        int f=pa,g=t[pa].fa,o=gi(x);
        if(g) t[g].ch[gi(f)]=x;pa=g;
        t[f].ch[o]=t[x].ch[!o];t[t[x].ch[!o]].fa=f;
        t[x].ch[!o]=f;t[f].fa=x;
        upd(f);upd(x);
    }
    inline void splay(int x,int tar)
    {
        for(;pa!=tar;rot(x))
            if(t[pa].fa!=tar)
                rot((gi(x)==gi(pa))?pa:x);
        if(!tar) root=x;
    }
    void ins(int p)
    {
        int x=root,f=0,v=t[p].val;
        while(x) f=x,x=(v<t[x].val?ls:rs);
        x=p;
        t[f].ch[v>t[f].val]=x;t[x].fa=f;
        splay(x,0);
    }
    void order(int x)
    {
        int l=ls,r=rs;
        ls=rs=pa=0;t[x].siz=1;
        if(l) order(l);
        ins(x);
        if(r) order(r);
    }
    void merge(int x,int y)
    {
        if(x==y) return;
        splay(x,0);splay(y,0);
        if(t[x].siz>t[y].siz) swap(x,y);
        fa[x]=y;root=y;
        order(x);
    }
    int kth(int x,int k)
    {
        splay(x,0);
        while(x)
        {
            if(k<=t[ls].siz) x=ls;
            else if(k>t[ls].siz+1) k-=t[ls].siz+1,x=rs;
            else return x;
        }
        return -1;
    }
    void print()
    {
        printf("%d\n",root);
        for(int x=1;x<=n;x++) printf("%d %d %d %d\n",ls,rs,pa,t[x].siz);
    }
}S;
char s[10];
int main()
{
//  freopen("in.in","r",stdin);
//  freopen("out.out","w",stdout);
    n=in(),m=in();
    for(int i=1;i<=n;i++) fa[i]=i,val[i]=in();
    S.init();
    for(int i=1,x,y;i<=m;i++) x=find(in()),y=find(in()),S.merge(x,y);
    m=in();
//  S.print();
    for(int i=1,x,y;i<=m;i++)
    {
        scanf("%s",s);
        if(s[0]=='B') S.merge(find(in()),find(in()));
        else x=in(),y=in(),printf("%d\n",S.kth(x,y));
//      S.print();
//      for(int i=1;i<=n;i++) cout<<fa[i]<<" ";
//      cout<<endl;
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值