私は Mocha!!

私はこの世界を深く愛している

[BZOJ3223]JoyOI1729 文艺平衡树(Splay)

传送门


题意就是区间翻转。

首先我们发现,一开始的序列是有序的,而且等于他的位置,同时不会修改,所以我们把在数组中的位置当作权值,使用splay进行维护。对于翻转l到r,我们首先把l-1旋转 到根,再把r+1旋转到根的右子树的根,那么根的右子树的左子树就是区间了l,r那么l到r就在一个子树里面了,我们就可以通过打翻转标记来进行翻转。


#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=110000;
struct trnode//以size作为关键字建立splay 
{
    int son[2],fz,c,f,d;
}tr[N*2]; int len,root;
void update(int x)
{
    if(x==0) return;
    int lc=tr[x].son[0],rc=tr[x].son[1];
    tr[x].c=tr[lc].c+tr[rc].c+1;
    if(tr[x].fz)
    {
        tr[x].fz^=1; tr[lc].fz^=1,tr[rc].fz^=1;
        swap(tr[x].son[0],tr[x].son[1]);
    }
    return ;
}
void add(int d,int f)
{
    len++; tr[len].fz=0;
    tr[len].son[0]=tr[len].son[1]=0;
    tr[len].c=1;tr[len].d=d;tr[len].f=f;
    if(d<tr[f].d) tr[f].son[0]=len;
    else tr[f].son[1]=len;
    return;
}
void rotate(int x)
{
    int w;
    int f=tr[x].f,ff=tr[f].f;
    int r,R;

    if(tr[f].son[0]==x) w=1;
    else w=0;

    R=f; r=tr[x].son[w]; 
    tr[R].son[1-w]=r;
    if(r!=0) tr[r].f=R;

    R=ff,r=x;
    tr[R].son[ (f==tr[ff].son[0])?0:1 ]=r;
    tr[r].f=R;

    R=x,r=f;
    tr[R].son[w]=r;
    tr[r].f=R;

    update(f);
    update(x);
}
void splay(int x,int rt)
{
    while(tr[x].f!=rt)
    {
        int f=tr[x].f,ff=tr[f].f;
        update(x); update(f); update(ff);
        if(ff==rt) rotate(x);
        else
        {
            if( (f==tr[ff].son[0])==(x==tr[f].son[0]) ) rotate(f);
            else rotate(x);
            rotate(x);
        }
    }
    if(rt==0) root=x;
}
int findip(int d)
{
    int x=root;
    while(tr[x].d!=d)
    {
        if(tr[x].d>d)
        {
            if(tr[x].son[0]) x=tr[x].son[0];
            else break;
        }
        else
        {
            if(tr[x].son[1]) x=tr[x].son[1];
            else break;
        }
    }
    return x;
}
int findpm(int k)
{
    int x=root;
    while(1)
    {
        update(x);
        int lc=tr[x].son[0],rc=tr[x].son[1];
        if(k<=tr[lc].c) x=lc;
        else if(k>tr[lc].c+1) {k-=tr[lc].c+1; x=rc;}
        else break;
    }
    splay(x,0);
    return tr[x].d;
}
void ins(int d)
{
    if(root==0) {add(d,0); root=len;return;}
    int x=findip(d);
    add(d,x);
    update(x);
    splay(x,0);
}
int pri[N],ll=0;
void print(int x)
{
    update(x);
    if(tr[x].son[0]!=0) print(tr[x].son[0]);
    pri[++ll]=tr[x].d;
    if(tr[x].son[1]!=0) print(tr[x].son[1]);
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) ins(i);
    for(int i=1;i<=m;i++)
    {
        int l,r; scanf("%d%d",&l,&r);
        //把l和r中间的转到一个子树去
        if(l!=1 && r!=n)
        {
            int p1=findpm(l-1);
            int p2=findpm(r+1); 
            splay(p1,0);
            splay(p2,root);
            tr[tr[p2].son[0]].fz^=1;
        }
        else if(l==1 && r!=n)
        { 
            int p=findpm(r+1);
            splay(p,0); 
            tr[tr[p].son[0]].fz^=1; 
        }
        else if(l!=1 && r==n)
        {
            int p=findpm(l-1);
            splay(p,0);
            tr[tr[p].son[1]].fz^=1; 
        }
        else tr[root].fz^=1;//1~n
    }
    print(root);
    for(int i=1;i<n;i++) printf("%d ",pri[i]);
    printf("%d\n",pri[n]);
    return 0; 
}
阅读更多
版权声明:本文为博主原创文章,转载请附上原博客链接。 https://blog.csdn.net/CABI_ZGX/article/details/79948869
个人分类: Splay
上一篇[BZOJ1997]Planar(2-SAT)
下一篇Fhq-Treap总结:短小精悍不旋转的神级数据结构
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭