1. 程式人生 > >P3721 [AH2017/HNOI2017]單旋

P3721 [AH2017/HNOI2017]單旋

dig [1] lib hnoi lin mes read sin 們的

題目:https://www.luogu.org/problemnew/show/P3721

手玩一下即可AC此題。
結論:插入x後,x要麽會成為x的前驅的右兒子,要麽成為x的後繼的左兒子,這取決於它的前驅和後繼的深度。
證明:首先可以證明的是,x的前驅和後繼一定存在祖先與後代的關系,因為如果不存在此關系,它們的LCA一定和雙方更接近。
然後這個結論畫畫圖就比較顯然了。
結論:單旋刪除最小值後,它連向根節點的這條路徑不發生變化,手玩即可證明,改變的只有它的兒子。

那麽這個題就顯然可以用LCT來維護,查詢深度也很簡單。

#include<iostream>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<set> 
#include<ctime>
#include<cstdlib>
#include<algorithm>
#define N 330000
#define L 300000
#define eps 1e-7
#define inf 1e9+7
#define db double
#define ll long long
#define ldb long double
using namespace std;
inline int read()
{
    char ch=0;
    int x=0,flag=1;
    while(!isdigit(ch)){ch=getchar();if(ch==‘-‘)flag=-1;}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-‘0‘;ch=getchar();}
    return x*flag;
}
#define lson son[x][0]
#define rson son[x][1]
int root,v[N],f[N],st[N],flag[N],sumv[N],son[N][2];
void pushup(int x){sumv[x]=sumv[lson]+sumv[rson]+1;}
void update(int x){flag[x]^=1;swap(lson,rson);}
void pushdown(int x){if(!flag[x])return;if(lson)update(lson);if(rson)update(rson);flag[x]=0;}
bool get(int x){return son[f[x]][1]==x;}
bool isroot(int x){return (son[f[x]][0]!=x)&&(son[f[x]][1]!=x);}
void rotate(int x)
{
    int y=f[x],z=f[y],tx=get(x),ty=get(y),p=son[x][!tx];
    if(!isroot(y))son[z][ty]=x;son[x][!tx]=y;son[y][tx]=p;
    if(p)f[p]=y;f[y]=x;f[x]=z;pushup(y);pushup(x);
}
void splay(int x)
{
    int cnt=0,tmp=x;
    st[++cnt]=x;
    while(!isroot(x))st[++cnt]=f[x],x=f[x];
    for(int i=cnt;i>=1;i--)pushdown(st[i]);
    x=tmp;
    while(!isroot(x))
    {
        int y=f[x];
        if(!isroot(y))rotate(get(x)==get(y)?y:x);
        rotate(x);
    }
    pushup(x);
}
void access(int x){for(int y=0;x;y=x,x=f[x])splay(x),son[x][1]=y,pushup(x);}
void makeroot(int x){access(x);splay(x);update(x);}
void link(int x,int y){if(!x||!y)return;makeroot(x);f[x]=y;}
void cut(int x,int y){if(!x||!y)return;makeroot(x);access(y);splay(y);f[x]=son[y][0]=0;pushup(y);}
int dep(int x){makeroot(root);access(x);splay(x);return sumv[x];}
set<int>S;
set<int>::iterator it;
int opt[N],w[N],W[N],fa[N],lc[N],rc[N];
int main()
{
    int m=read(),num=0;
    for(int i=1;i<=m;i++)
    {
        opt[i]=read();
        if(opt[i]==1)w[i]=W[++num]=read();
    }
    sort(W+1,W+num+1);num=unique(W+1,W+num+1)-W-1;
    for(int i=1;i<=m;i++)if(opt[i]==1)w[i]=lower_bound(W+1,W+num+1,w[i])-W;
    for(int i=1;i<=m;i++)
    {
        lc[0]=rc[0]=fa[0]=fa[root]=0;
        int flag=opt[i];
        if(flag==1)
        {
            int x=w[i],p,q;
            S.insert(x);it=S.find(x);
            if(it==S.begin())p=-1;else p=*(--S.lower_bound(x));
            if(it==(--S.end()))q=-1;else q=*(S.upper_bound(x));
            if(p==-1&&q==-1){root=x;printf("1\n");continue;}
            if(p==-1||dep(p)<dep(q))link(x,q),lc[q]=x,fa[x]=q;
            else link(x,p),rc[p]=x,fa[x]=p;
            printf("%d\n",dep(x));
        }
        if(flag==2||flag==4)
        {
            int x=*S.begin(),p=rc[x],q=fa[x];
            printf("%d\n",dep(x));
            if(x!=root)
            {
                cut(x,p);cut(x,q);link(p,q);link(x,root);
                lc[q]=p;if(p)fa[p]=q;rc[x]=root;fa[x]=0;fa[root]=x;root=x;
            }
        }
        if(flag==3||flag==5)
        {
            int x=*S.rbegin(),p=lc[x],q=fa[x];
            printf("%d\n",dep(x));
            if(x!=root)
            {
                cut(x,p);cut(x,q);link(p,q);link(x,root);
                rc[q]=p;if(p)fa[p]=q;lc[x]=root;fa[x]=0;fa[root]=x;root=x;
            }
        }
        if(flag==4)
        {
            int x=*S.begin(),p=rc[x];
            if(p)cut(x,p),rc[x]=fa[p]=0;
            root=p;S.erase(x);
        }
        if(flag==5)
        {
            int x=*S.rbegin(),p=lc[x];
            if(p)cut(x,p),lc[x]=fa[p]=0;
            root=p;S.erase(x);
        }
    }
    return 0;
}

P3721 [AH2017/HNOI2017]單旋