[BZOJ1500][NOI2005]维修数列(平衡树)

题目:

我是超链接

题解:

md这题调的我哭出来,最后发现是all数组del的时候没清零Orz
反正是平衡树的基本操作吧
因为害怕转着转着出现奇怪的错误,我们把1和n+2当做空节点,权值为-INF,但是sum,maxx都是0,避免影响结果,注意这里0节点也要处理一下,因为我们用ch[now][0/1]的时候节点也是0,容易GG
Insert:只需把pos点splay到根,pos+1splay到根的右儿子,因为pos+1是pos的后继,因此pos+1一定没有左儿子,所以新建一棵树,然后再把新建成的树挂在pos+1左儿子的位置上。
Delete:只需把pos-1旋转到根,pos+tot旋转到根的右子节点,删除pos+tot的左儿子
Make-same:把pos-1 Splay到根,把pos+tot Splay到根节点右边,加一个all的标记
Reverse:把pos-1转到根,pos+tot转到根节点右边,将pos+tot的reverse标记取反。
Get-sum:把pos-1转到根,pos+tot转到根的右边,输出pos+tot左儿子的totsum。
Max-sum:把MIN转到根,MAX转到MIN的右边,输出MAX左儿子的maxsum值。

这里要求最大连续子段和,线段树似乎做过吧,维护左开始最大,右开始最大,总的最大然后合并就好了

代码:

#include 
#include 
#define INF 1e9
using namespace std;
const int N=500005;
int ch[N][2],stack[N],top,sz,sum[N],size[N],maxx[N],lm[N],rm[N],all[N],delta[N],f[N],key[N],a[N],root;
void init(int now,int v)
{
    ch[now][0]=ch[now][1]=all[now]=delta[now]=f[now]=0;////
    lm[now]=rm[now]=maxx[now]=sum[now]=key[now]=v; 
}
int get(int x){return ch[f[x]][1]==x;}
int neww(){return (top)?stack[top--]:++sz;}
void updata(int now)
{
    int l=ch[now][0],r=ch[now][1];
    sum[now]=sum[l]+sum[r]+key[now];
    size[now]=size[l]+size[r]+1;
    maxx[now]=max(max(maxx[l],maxx[r]),max(0,lm[r])+key[now]+max(0,rm[l]));
    lm[now]=max(lm[l],sum[l]+key[now]+max(0,lm[r]));
    rm[now]=max(rm[r],sum[r]+key[now]+max(0,rm[l]));
}
void cover(int now,int v)
{
    sum[now]=v*size[now]; key[now]=v;
    all[now]=1; delta[now]=0;
    maxx[now]=lm[now]=rm[now]=max(sum[now],key[now]);
}
void fz(int now)
{
    if (all[now]) return;
    delta[now]^=1;
    swap(ch[now][0],ch[now][1]);
    swap(lm[now],rm[now]); 
}
void pushdown(int now)
{
    int l=ch[now][0],r=ch[now][1];
    if (delta[now])
    {
        if (l) fz(l);
        if (r) fz(r);
        delta[now]=0;
    }
    if (all[now])
    {
        if (l) cover(l,key[now]);
        if (r) cover(r,key[now]);
        all[now]=0;
    }
}
void rotate(int x)
{
    pushdown(f[x]);pushdown(x);
    int old=f[x],oldf=f[old],which=get(x);
    f[x]=oldf; if(oldf) ch[oldf][ch[oldf][1]==old]=x;
    ch[old][which]=ch[x][which^1]; f[ch[x][which^1]]=old;
    ch[x][which^1]=old; f[old]=x;
    updata(old);
    updata(x);
}
void splay(int x,int to)
{
    for (;f[x]!=to;rotate(x))
      if (f[f[x]]!=to) rotate(get(x)==get(f[x])?f[x]:x); 
    if (!to) root=x;
}
int build(int fa,int l,int r)
{
    if (l>r) return 0;
    int mid=(l+r)>>1;
    int now=neww(); init(now,a[mid]);////
    f[now]=fa;
    int lch=build(now,l,mid-1);
    int rch=build(now,mid+1,r);
    ch[now][0]=lch; ch[now][1]=rch;
    updata(now);
    return now;
}
int find(int x)
{
    int now=root;
    while (1)
    {
        pushdown(now);
        if (x<=size[ch[now][0]]) now=ch[now][0];
        else 
        {
            x-=size[ch[now][0]]+1;
            if (!x) return now;
            now=ch[now][1];
        }
    }
}
void insert()
{
    int pos,tot,x,y;
    scanf("%d%d",&pos,&tot);
    for (int i=1;i<=tot;i++) scanf("%d",&a[i]);
    x=find(pos+1); splay(x,0);
    y=find(pos+2); splay(y,x);
    int j=build(y,1,tot); ch[y][0]=j;
    updata(y); updata(x);
}
void clear(int now)
{
    if (!now) return;
    stack[++top]=now;
    clear(ch[now][0]); clear(ch[now][1]);
}
void del()
{
    int pos,tot,x,y;
    scanf("%d%d",&pos,&tot);    
    x=find(pos); splay(x,0);
    y=find(pos+tot+1); splay(y,x);
    clear(ch[y][0]); ch[y][0]=0;
    updata(y); updata(x);
}
void same()
{
    int pos,tot,x,y,c;
    scanf("%d%d%d",&pos,&tot,&c);
    x=find(pos); splay(x,0);
    y=find(pos+tot+1); splay(y,x);
    cover(ch[y][0],c);
    updata(y); updata(x);
}
void rev()
{
    int pos,tot,x,y;
    scanf("%d%d",&pos,&tot);
    x=find(pos); splay(x,0);
    y=find(pos+tot+1); splay(y,x);
    fz(ch[y][0]);
    updata(y); updata(x);
}
int getsum()
{
    int pos,tot,x,y;
    scanf("%d%d",&pos,&tot);
    x=find(pos); splay(x,0);
    y=find(pos+tot+1); splay(y,x);
    return sum[ch[y][0]];
}
int main()
{
    int n,m;scanf("%d%d",&n,&m);
    for (int i=2;i<=n+1;i++) scanf("%d",&a[i]);a[1]=a[n+2]=-INF;
    init(0,-INF); sum[0]=size[0]=0;////
    root=build(0,1,n+2); 
    for (int i=1;i<=m;i++)
    {
        char st[15];
        scanf("%s",st);
        if (st[2]=='X') printf("%d\n",maxx[root]);
        else if (st[0]=='I') insert();
        else if (st[0]=='D') del();
        else if (st[0]=='M') same();
        else if (st[0]=='R') rev();
        else if (st[0]=='G') printf("%d\n",getsum());
    }
}

你可能感兴趣的:(平衡树)