luogu2042 维护数列

http://www.elijahqi.win/archives/1692
题目描述
请写一个程序,要求维护一个数列,支持以下 6 种操作:(请注意,格式栏 中的下划线‘ _ ’表示实际输入文件中的空格)

输入输出格式
输入格式:

输入文件的第 1 行包含两个数 N 和 M,N 表示初始时数列中数的个数,M 表示要进行的操作数目。 第 2 行包含 N 个数字,描述初始时的数列。 以下 M 行,每行一条命令,格式参见问题描述中的表格

输出格式:

对于输入数据中的 GET-SUM 和 MAX-SUM 操作,向输出文件依次打印结 果,每个答案(数字)占一行。

输入输出样例
输入样例#1: 复制

9 8 2 -6 3 5 1 -5 -3 6 3
GET-SUM 5 4
MAX-SUM
INSERT 8 3 -5 7 2
DELETE 12 1
MAKE-SAME 3 3 2
REVERSE 3 6
GET-SUM 5 4
MAX-SUM
输出样例#1: 复制

-1
10
1
10
说明
你可以认为在任何时刻,数列中至少有 1 个数。

输入数据一定是正确的,即指定位置的数在数列中一定存在。

50%的数据中,任何时刻数列中最多含有 30 000 个数;

100%的数据中,任何时刻数列中最多含有 500 000 个数。

100%的数据中,任何时刻数列中任何一个数字均在[-1 000, 1 000]内。

100%的数据中,M ≤20 000,插入的数字总数不超过 4 000 000 。

平衡树 恶心大模板 以前写过的

每天晚上回宿舍刚 然而 两个晚上过去到今晚才调完 还是我太菜了 可以看看网上的题解 关于左旋右旋 还有伸展操作的讲解

我这个似乎是黄学长的代码 很短 好想好记

#include<queue>
#include<cstdio>
#include<algorithm>
#define inf 0x3f3f3f3f
#define N 550000
using namespace std;
queue<int>q;
int father[N],child[N][2],value[N],size[N],sum[N],w[N],id[N],maxl[N],maxr[N],maxs[N],cnt,root,n,m;
bool tag[N],rev[N];char s[20];
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void print(int x){
    if (child[x][0]) print(child[x][0]);
    printf("value size sum maxs maxl maxr father x childleft childright\n");
    printf("%d %d %d %d %d %d %d %d %d %d\n",value[x],size[x],sum[x],maxs[x],maxl[x],maxr[x],father[x],x,child[x][0],child[x][1]);
    if (child[x][1]) print(child[x][1]);
}
inline void pushdown(int x){
    int l=child[x][0],r=child[x][1];
    if (tag[x]){
        tag[x]=rev[x]=0;
        if (l) tag[l]=1,value[l]=value[x],sum[l]=size[l]*value[x];
        if (r) tag[r]=1,value[r]=value[x],sum[r]=size[r]*value[x];
        if (value[x]>=0){
            if (l) maxl[l]=maxr[l]=maxs[l]=sum[l];
            if (r) maxl[r]=maxr[r]=maxs[r]=sum[r];
        } else {
            if(l) maxl[l]=maxr[l]=0,maxs[l]=value[l];if (r) maxl[r]=maxr[r]=0,maxs[r]=value[r];
        }
    }
    if (rev[x]){
        rev[x]=0;rev[l]^=1;rev[r]^=1;
        swap(maxl[l],maxr[l]);swap(maxl[r],maxr[r]);
        swap(child[l][0],child[l][1]);swap(child[r][0],child[r][1]);
    }
}
inline int find(int x,int rk){
    pushdown(x);int l=child[x][0],r=child[x][1];
    if (size[l]+1==rk) return x;
    if (size[l]>=rk) return find(l,rk);else return find(r,rk-size[l]-1);
}
inline void update(int x){
    int l=child[x][0],r=child[x][1];
    sum[x]=sum[l]+sum[r]+value[x];
    size[x]=size[l]+size[r]+1;
    maxs[x]=max(maxs[l],maxs[r]);
    maxs[x]=max(maxs[x],maxr[l]+value[x]+maxl[r]);
    maxr[x]=max(maxr[r],sum[r]+value[x]+maxr[l]);
    maxl[x]=max(maxl[l],sum[l]+value[x]+maxl[r]);
}
void build (int fa,int l,int r){
    if (l>r) return ;
    int mid=l+r>>1,now=id[mid],last=id[fa];
    if (l==r){
        size[now]=1;sum[now]=w[l];tag[now]=rev[now]=0;
        if (w[l]>=0) maxl[now]=maxr[now]=maxs[now]=w[l];
        else maxl[now]=maxr[now]=0,maxs[now]=w[l];
    }else build(mid,l,mid-1),build(mid,mid+1,r);
    value[now]=w[mid];father[now]=last;update(now);child[last][mid>=fa]=now;
}
inline void rotate(int x,int &tar){
    int y=father[x],z=father[y],l,r;
    l=(child[y][1]==x),r=l^1;
    if (y==tar) tar=x;else child[z][child[z][1]==y]=x;
    father[child[x][r]]=y;father[y]=x;father[x]=z;
    child[y][l]=child[x][r];child[x][r]=y;
    update(y);update(x); 
}
inline void splay(int x,int &tar){
    while(x!=tar){
        int y=father[x],z=father[y];
        if (y!=tar)
            if (child[y][0]==x^child[z][0]==y) rotate(x,tar);else rotate(y,tar);
        rotate(x,tar);
    }
}
inline int split(int posi,int tot){
    int p1=find(root,posi),p2=find(root,posi+tot+1);
    splay(p2,root);splay(p1,child[p2][0]);return child[p1][1];
}
inline int gs(int pos,int tot,int id){int p=split(pos,tot);return sum[p];}
inline void insert1(int posi,int tot){
    for (int i=1;i<=tot;++i)
        if (!q.empty()) id[i]=q.front(),q.pop();else id[i]=++cnt;
    build(0,1,tot);
    int mid=id[1+tot>>1],p1=find(root,posi+1),p2=find(root,posi+2);
    splay(p2,root);splay(p1,child[root][0]);father[mid]=p1;child[p1][1]=mid;
    update(p1);update(p2);
}
inline void modify(int x,int posi,int tot,int c){
    int tmp=split(posi,tot);value[tmp]=c;tag[tmp]=1;sum[tmp]=c*size[tmp];
    if (c>=0) maxl[tmp]=maxr[tmp]=maxs[tmp]=sum[tmp];
    else maxl[tmp]=maxr[tmp]=0,maxs[tmp]=value[tmp];
    update(father[tmp]);update(father[father[tmp]]);
}
inline void rec(int x){
    if (x==0) return;
    rec(child[x][0]);rec(child[x][1]);
    q.push(x);father[x]=child[x][0]=child[x][1]=0;tag[x]=rev[x]=0;
}
inline void delete1(int posi,int tot){
    int tmp=split(posi,tot);int fa=father[tmp];
    rec(tmp);child[fa][1]=0;update(fa);update(father[fa]);
}
inline void reverse(int posi,int tot){
    int tmp=split(posi,tot);int fa=father[tmp];
    if (!tag[tmp]){
        rev[tmp]^=1;
        swap(child[tmp][0],child[tmp][1]);swap(maxl[tmp],maxr[tmp]);
        update(fa);update(father[fa]);
    }
}
int main(){
    //freopen("2042.in","r",stdin);
//  freopen("2042.out","w",stdout);
    n=read();m=read();
    for (int i=1;i<=n;++i) w[i+1]=read();
    for (int i=1;i<=n+2;++i) id[i]=i;
    maxs[0]=w[1]=w[n+2]=-inf;
    build(0,1,n+2); cnt=n+2;root=(1+cnt)>>1;
    for (int i=1;i<=m;++i){
        scanf("%s",s);
        if (s[2]=='S'){
            int posi=read(),tot=read();
            for (int i=1;i<=tot;++i) w[i]=read();insert1(posi,tot);
        }
        if (s[2]=='T'){
            int pos=read(),tot=read();
            int tmp=gs(pos,tot,i);
            //printf("%d %d\n",pos,tot);
            printf("%d\n",tmp);
        }
        if (s[2]=='X') printf("%d\n",maxs[root]);
        if (s[2]=='K'){
            int posi=read(),tot=read(),c=read();
            modify(root,posi,tot,c);
        }
        if (s[2]=='L'){
            int posi=read(),tot=read();delete1(posi,tot);
        }
        if (s[2]=='V'){
            int posi=read(),tot=read();reverse(posi,tot);
        }
    }
    return 0;   
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值