AC日记——[HNOI2012]永无乡 bzoj 2733

2733

思路：

　　启发式合并splay（n*log^2n）；

来，上代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
 
using namespace std;
 
#define maxn 100001
 
int n,q,tot,ch[maxn*][],key[maxn*],w[maxn*],opi[maxn*],m;
int size[maxn*],root[maxn],id[maxn*],f[maxn],cnt,dis[maxn*],lar[maxn];
 
inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'') Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}
 
inline int getson(int now)
{
    return ch[opi[now]][]==now;
}
 
inline void updata(int now)
{
    size[now]=w[now];
    if(ch[now][]) size[now]+=size[ch[now][]];
    if(ch[now][]) size[now]+=size[ch[now][]];
}
 
inline void rotate(int now)
{
    int fa=opi[now],ffa=opi[fa],pos=getson(now);
    ch[fa][pos]=ch[now][pos^];
    if(ch[fa][pos]) opi[ch[fa][pos]]=fa;
    if(ffa) ch[ffa][getson(fa)]=now;
    ch[now][pos^]=fa,opi[fa]=now,opi[now]=ffa;
    updata(fa),updata(now);
}
 
inline void splay(int now,int to)
{
    for(int fa;fa=opi[now];rotate(now))
    {
        if(opi[fa]) rotate(getson(now)==getson(fa)?fa:now);
    }
    root[to]=now;
}
 
inline void insert(int x,int y,int to)
{
    if(!root[to])
    {
        root[to]=++tot,key[tot]=x,w[tot]=size[tot]=,id[tot]=y;
        return ;
    }
    int now=root[to],fa=;
    while()
    {
        fa=now;
        if(x<key[now]) now=ch[now][];
        else now=ch[now][];
        if(!now)
        {
            now=ch[fa][x>key[fa]]=++tot;
            key[now]=x,id[now]=y,w[now]=size[now]=,opi[now]=fa;
            splay(now,to);break;
        }
    }
}
 
inline int find(int x)
{
    if(x==f[x]) return x;
    return f[x]=find(f[x]);
}
 
inline int irank(int k,int to)
{
    int now=root[to];
    while()
    {
        int dis=size[ch[now][]];
        if(k<=dis) now=ch[now][];
        else
        {
            k-=dis;
            if(k<=w[now])
            {
                splay(now,to);
                return id[now];
            }
            else k-=w[now],now=ch[now][];
        }
    }
}
 
inline void merge(int noww,int now)
{
    if(!now) return ;
    if(ch[now][]) merge(noww,ch[now][]);
    insert(key[now],id[now],noww);
    if(ch[now][]) merge(noww,ch[now][]);
}
 
int main()
{
    in(n),in(m);int x,y;char ch[];
    for(int i=;i<=n;i++) in(dis[i]),f[i]=i,insert(dis[i],i,i),lar[i]=;
    for(;m--;)
    {
        in(x),in(y);
        x=find(x),y=find(y);
        if(lar[x]<lar[y]) swap(x,y);lar[x]+=lar[y];
        if(x!=y) f[y]=x,merge(x,root[y]);
    }
    in(q);
    for(;q--;)
    {
        scanf("%s",ch);in(x),in(y);
        if(ch[]=='Q')
        {
            x=find(x);
            if(y<=size[root[x]]) printf("%d\n",irank(y,x));
            else printf("-1\n");
        }
        else
        {
            x=find(x),y=find(y);
            if(lar[x]<lar[y]) swap(x,y);lar[x]+=lar[y];
            if(x!=y) f[y]=x,merge(x,root[y]);
        }
    }
    return ;
}