[NOI2005]维修数列(SplayTree)

来源：互联网发布：ping测试网络编辑：程序博客网时间：2024/04/27 15:56

[NOI2005]维修数列(SplayTree)

http://www.cnblogs.com/kuangbin/archive/2013/04/21/3034081.html

分析:

本题看起来复杂,其实和之前做的SplayTree基本一样就是多了个维护最大连续和的.最大连续和维护需要最大前缀和lx和最大后缀和rx.(与之前的线段树维护最大连续和基本一样)

具体实现看代码.

注意:setv[i]=1,表示i节点的树要执行set操作,setv[i]=0表示不用执行set操作.执行set操作的时候,直接用key[i]的值去set.(因为这里面set操作有可能要set 0 ,所以0不能用来表示set操作不存在)

AC代码:

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int maxn=500000+100;const int INF=0x3f3f3f3f;#define Key_Value ch[ch[root][1]][0]int pre[maxn],ch[maxn][2],size[maxn],key[maxn],root,tot1;int sum[maxn],rev[maxn],setv[maxn],lx[maxn],rx[maxn],mx[maxn];int s[maxn],tot2;int n,q;int a[maxn];void New_Node(int &r,int fa,int k){    if(tot2) r=s[tot2--];    else r=++tot1;    key[r]=k;    pre[r]=fa;    rev[r]=setv[r]=ch[r][0]=ch[r][1]=0;    size[r]=1;    sum[r]=lx[r]=rx[r]=mx[r]=k;}void Update_Rev(int r){    if(!r) return ;    swap(ch[r][0],ch[r][1]);    swap(lx[r],rx[r]);    rev[r]^=1;}void Update_Set(int r,int val){    if(!r) return;    setv[r]=1;//注意    key[r]=val;    sum[r]=size[r]*val;    lx[r]=rx[r]=mx[r]=max(val,val*size[r]);}void Push_Up(int r){    int lson=ch[r][0],rson=ch[r][1];    size[r]=1+size[lson]+size[rson];    sum[r]=key[r]+sum[lson]+sum[rson];    lx[r]=max(lx[lson],sum[lson]+key[r]+max(0,lx[rson]));    rx[r]=max(rx[rson],sum[rson]+key[r]+max(0,rx[lson]));    mx[r]=max(0,rx[lson])+key[r]+max(0,lx[rson]);    mx[r]=max(mx[r],max(mx[lson],mx[rson]));}void Push_Down(int r){    if(setv[r])    {        Update_Set(ch[r][0],key[r]);        Update_Set(ch[r][1],key[r]);        setv[r]=0;    }    if(rev[r])    {        Update_Rev(ch[r][0]);        Update_Rev(ch[r][1]);        rev[r]=0;    }}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][0]=ch[root][1]=size[root]=sum[root]=rev[root]=setv[root]=key[root]=0;    lx[root]=rx[root]=mx[root]=-INF;    New_Node(root,0,-1);    New_Node(ch[root][1],root,-1);    for(int i=0;i<n;i++)scanf("%d",&a[i]);    Build(Key_Value,0,n-1,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]]+1;    if(k==t) return r;    else if(k<t) return Get_Kth(ch[r][0],k);    else return Get_Kth(ch[r][1],k-t);}void Insert(int pos,int tot){    for(int i=0;i<tot;i++) scanf("%d",&a[i]);    Splay(Get_Kth(root,pos+1),0);    Splay(Get_Kth(root,pos+2),root);    Build(Key_Value,0,tot-1,ch[root][1]);    Push_Up(ch[root][1]);    Push_Up(root);}void erase(int r){    if(!r) return;    s[++tot2]=r;    erase(ch[r][0]);    erase(ch[r][1]);}void Delete(int pos,int tot){    Splay(Get_Kth(root,pos),0);    Splay(Get_Kth(root,pos+tot+1),root);    erase(Key_Value);    pre[Key_Value]=0;    Key_Value=0;    Push_Up(ch[root][1]);    Push_Up(root);}void Set(int pos,int tot,int c){    Splay(Get_Kth(root,pos),0);    Splay(Get_Kth(root,pos+tot+1),root);    Update_Set(Key_Value,c);    Push_Up(ch[root][1]);    Push_Up(root);}void Reverse(int pos,int tot){    Splay(Get_Kth(root,pos),0);    Splay(Get_Kth(root,pos+tot+1),root);    Update_Rev(Key_Value);    Push_Up(ch[root][1]);    Push_Up(root);}int Get_Sum(int pos,int tot){    Splay(Get_Kth(root,pos),0);    Splay(Get_Kth(root,pos+tot+1),root);    return sum[Key_Value];}int Get_MaxSum(int pos,int tot){    Splay(Get_Kth(root,pos),0);    Splay(Get_Kth(root,pos+tot+1),root);    return mx[Key_Value];}int main(){    while(scanf("%d%d",&n,&q)==2)    {        Init();        char op[20];        int x,y,z;        while(q--)        {            scanf("%s",op);            if(op[0]=='I')            {                scanf("%d%d",&x,&y);                Insert(x,y);            }            else if(op[0]=='D')            {                scanf("%d%d",&x,&y);                Delete(x,y);            }            else if(op[0]=='M'&&op[2]=='K')            {                scanf("%d%d%d",&x,&y,&z);                Set(x,y,z);            }            else if(op[0]=='R')            {                scanf("%d%d",&x,&y);                Reverse(x,y);            }            else if(op[0]=='G')            {                scanf("%d%d",&x,&y);                printf("%d\n",Get_Sum(x,y));            }            else            {                printf("%d\n",Get_MaxSum(1,size[root]-2));            }        }    }    return 0;}

0 0