POJ 3580 SuperMemo(SplayTree)

POJ 3580 SuperMemo(SplayTree)

http://poj.org/problem?id=3580

题意:要求你维护一个数列,完成下面6种操作,区间加,区间翻转,区间循环移位,插入,删除,区间求最小值.

分析:

除了循环移位外,其他都是SplayTree的基本操作,现在来考虑循环移位.区间[l,r]循环右移T(T=T%(r-l+1) )位,等于把区间[l,r-T]和区间[r-T+1,r]调换了位置.

由于需要实现删除操作,所以要有内存回收的机制.

注意:这里的删除操作,由于我们每个节点都维护了信息,如果你要直接通过赋值删除的话,有可能父节点的addv或者rev信息没有传下去,导致出错,所以如果想通过赋值删除,一定要注意Push_Down操作先.

AC代码:

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define Key_Value ch[ch[root][1]][0]
const int INF=0x3f3f3f3f;
const int maxn=200000+100;
int pre[maxn],size[maxn],ch[maxn][2],addv[maxn],minv[maxn],key[maxn],rev[maxn],root,tot1;
int s[maxn],tot2;
int a[maxn];
int n,q;

void Update_Add(int r,int add)
{
    if(r==0) return;
    key[r]+=add;
    minv[r]+=add;
    addv[r]+=add;
}
void Update_Rev(int r)
{
    if(r==0) return;
    swap(ch[r][0],ch[r][1]);
    rev[r]^=1;
}
void Push_Down(int r)
{
    if(rev[r])
    {
        Update_Rev(ch[r][0]);
        Update_Rev(ch[r][1]);
        rev[r]=0;
    }
    if(addv[r])
    {
        Update_Add(ch[r][0],addv[r]);
        Update_Add(ch[r][1],addv[r]);
        addv[r]=0;
    }
}
void Push_Up(int r)
{
    size[r]=1+size[ch[r][0]]+size[ch[r][1]];
    minv[r]=key[r];
    if(ch[r][0]) minv[r]=min(minv[r],minv[ch[r][0]]);
    if(ch[r][1]) minv[r]=min(minv[r],minv[ch[r][1]]);
}
void New_Node(int &r,int fa,int k)
{
    if(tot2) r=s[tot2--];
    else r=++tot1;
    minv[r]=key[r]=k;
    pre[r]=fa;
    rev[r]=addv[r]=ch[r][0]=ch[r][1]=0;
    size[r]=1;
}
void Build(int &x,int l,int r,int fa)
{
    if(l>r)return ;
    int mid=(l+r)>>1;
    New_Node(x,fa,a[mid]);
    Build(ch[x][0],l,mid-1,x);
    Build(ch[x][1],mid+1,r,x);
    Push_Up(x);
}
void init()
{
    root=tot1=tot2=0;
    pre[root]=ch[root][1]=ch[root][0]=addv[root]=key[root]=rev[root]=size[root]=0;
    minv[root]=INF;
    New_Node(root,0,INF);
    New_Node(ch[root][1],root,INF);
    Build(Key_Value,1,n,ch[root][1]);
    Push_Up(ch[root][1]);
    Push_Up(root);
}
void Rotate(int x,int kind)
{
    int y=pre[x];
    Push_Down(y);
    Push_Down(x);
    ch[y][!kind]=ch[x][kind];
    pre[ch[x][kind]]=y;
    if(pre[y]) ch[pre[y]][ch[pre[y]][1]==y] = x;
    pre[x]=pre[y];
    ch[x][kind]=y;
    pre[y]=x;
    Push_Up(y);
}
void Splay(int r,int goal)
{
    while(pre[r]!=goal)
    {
        if(pre[pre[r]]==goal)
            Rotate(r,ch[pre[r]][0]==r);
        else
        {
            int y=pre[r];
            int kind=ch[pre[y]][0]==y;
            if(ch[y][kind]==r)
            {
                Rotate(r,!kind);
                Rotate(r,kind);
            }
            else
            {
                Rotate(y,kind);
                Rotate(r,kind);
            }
        }
    }
    Push_Up(r);
    if(goal==0) root=r;
}
int Get_Kth(int r,int k)
{
    Push_Down(r);
    int t=size[ch[r][0]];
    if(t+1==k) return r;
    else if(k<=t) return Get_Kth(ch[r][0],k);
    else return Get_Kth(ch[r][1],k-1-t);
}
int Get_Max(int r)
{
    Push_Down(r);
    while(ch[r][1])
    {
        r=ch[r][1];
        Push_Down(r);
    }
    return r;
}
int Get_Min(int r)
{
    Push_Down(r);
    while(ch[r][0])
    {
        r=ch[r][0];
        Push_Down(r);
    }
    return r;
}
//下面是本题的操作
void ADD(int l,int r,int val)
{
    Splay(Get_Kth(root,l),0);
    Splay(Get_Kth(root,r+2),root);
    Update_Add(Key_Value,val);
    Push_Up(ch[root][1]);
    Push_Up(root);
}
void REVERSE(int l,int r)
{
    Splay(Get_Kth(root,l),0);
    Splay(Get_Kth(root,r+2),root);
    Update_Rev(Key_Value);
    Push_Up(ch[root][1]);
    Push_Up(root);
}
void REVOLVE(int l,int r,int T)//注意这里的写法,SplayTree删除一个区间和插入一个区间是如何实现的,并且不破坏本身的add和rev标记
{
    int len=r-l+1;
    T=(T%len);
    if(T==0) return ;
    int mid=r-T;//区间[l,mid]与[mid+1,r]调换位置
    Splay(Get_Kth(root,mid+1),0);
    Splay(Get_Kth(root,r+2),root);
    int temp=Key_Value;
    Key_Value=0;
    Push_Up(ch[root][1]);
    Push_Up(root);

    Splay(Get_Kth(root,l),0);
    Splay(Get_Kth(root,l+1),root);
    Key_Value=temp;
    pre[temp]=ch[root][1];
    Push_Up(ch[root][1]);
    Push_Up(root);
}
void INSERT(int x,int p)
{
    Splay(Get_Kth(root,x+1),0);
    Splay(Get_Kth(root,x+2),root);
    New_Node(Key_Value,ch[root][1],p);
    Push_Up(ch[root][1]);
    Push_Up(root);
}
void erase(int r)
{
    if(r)
    {
        s[++tot2]=r;
        erase(ch[r][0]);
        erase(ch[r][1]);
    }
}
void DELETE(int x)
{
    Splay(Get_Kth(root,x),0);
    Splay(Get_Kth(root,x+2),root);
    erase(Key_Value);
    pre[Key_Value]=0;
    Key_Value=0;
    Push_Up(ch[root][1]);
    Push_Up(root);
}
int MIN(int l,int r)
{
    Splay(Get_Kth(root,l),0);
    Splay(Get_Kth(root,r+2),root);
    return minv[Key_Value];
}
int main()
{
    int x,y,z;
    char op[20];
    while(scanf("%d",&n)==1)
    {
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        init();
        scanf("%d",&q);
        while(q--)
        {
            scanf("%s",op);
            if(strcmp(op,"ADD")==0)
            {
                scanf("%d%d%d",&x,&y,&z);
                ADD(x,y,z);
            }
            else if(strcmp(op,"REVERSE")==0)
            {
                scanf("%d%d",&x,&y);
                REVERSE(x,y);
            }
            else if(strcmp(op,"REVOLVE")==0)
            {
                scanf("%d%d%d",&x,&y,&z);
                REVOLVE(x,y,z);
            }
            else if(strcmp(op,"INSERT")==0)
            {
                scanf("%d%d",&x,&y);
                INSERT(x,y);
            }
            else if(strcmp(op,"DELETE")==0)
            {
                scanf("%d",&x);
                DELETE(x);
            }
            else if(strcmp(op,"MIN")==0)
            {
                scanf("%d%d",&x,&y);
                printf("%d\n",MIN(x,y));
            }
        }
    }
    return 0;
}