Description
请写一个程序,要求维护一个数列,支持以下 6 种操作:
请注意,格式栏 中的下划线‘ _ ’表示实际输入文件中的空格
Input
输入的第1 行包含两个数N 和M(M ≤20 000),N 表示初始时数列中数的个数,M表示要进行的操作数目。
第2行包含N个数字,描述初始时的数列。
以下M行,每行一条命令,格式参见问题描述中的表格。
任何时刻数列中最多含有500 000个数,数列中任何一个数字均在[-1 000, 1 000]内。
插入的数字总数不超过4 000 000个,输入文件大小不超过20MBytes。
Output
对于输入数据中的GET-SUM和MAX-SUM操作,向输出文件依次打印结果,每个答案(数字)占一行。
Sample Input
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
Sample Output
-1
10
1
10
HINT
解题思路:
考前敲模板,注意的地方:
1.newnode、update后记得return;
2.split后记得merge;
3.reverse是lmx和rmx也要交换。
#include<bits/stdc++.h>
using namespace std;
int getint()
{
int i=0,f=1;char c;
for(c=getchar();c!='-'&&(c<'0'||c>'9');c=getchar());
if(c=='-')f=-1,c=getchar();
for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
return i*f;
}
void print(int x)
{
if(x<0)putchar('-'),x=-x;
if(x>9)print(x/10);
putchar(x%10+'0');
}
const int N=500005,INF=1e9;
int n,m,a[N];
int root,tot,pool_top,pool[N],son[N][2],pri[N],val[N],sum[N],size[N],lmx[N],rmx[N],mx[N],rev[N],ms[N];
inline int newnode(int v)
{
int x=pool_top?pool[pool_top--]:++tot;
son[x][0]=son[x][1]=0,pri[x]=rand();
val[x]=sum[x]=mx[x]=v,size[x]=1;
lmx[x]=rmx[x]=max(v,0);
rev[x]=0,ms[x]=INF;
return x;
}
inline void rever(int x)
{
if(!x)return;
swap(son[x][0],son[x][1]),swap(lmx[x],rmx[x]),rev[x]^=1;
}
inline void cover(int x,int v)
{
val[x]=ms[x]=v,sum[x]=size[x]*v;
if(v>=0)lmx[x]=rmx[x]=mx[x]=size[x]*v;
else lmx[x]=rmx[x]=0,mx[x]=v;
}
inline void pushdown(int x)
{
int l=son[x][0],r=son[x][1];
if(rev[x])rever(l),rever(r),rev[x]=0;
if(ms[x]!=INF)cover(l,ms[x]),cover(r,ms[x]),ms[x]=INF;
}
inline int upt(int x)
{
int l=son[x][0],r=son[x][1];
sum[x]=sum[l]+val[x]+sum[r];
size[x]=size[l]+1+size[r];
lmx[x]=max(lmx[l],sum[l]+val[x]+lmx[r]);
rmx[x]=max(rmx[r],sum[r]+val[x]+rmx[l]);
mx[x]=max(rmx[l]+val[x]+lmx[r],max(mx[l],mx[r]));
return x;
}
int merge(int u,int v)
{
if(!v)return u;
if(!u)return v;
if(pri[u]<=pri[v])
{
pushdown(u),son[u][1]=merge(son[u][1],v);
return upt(u);
}
else
{
pushdown(v),son[v][0]=merge(u,son[v][0]);
return upt(v);
}
}
void split(int u,int k,int &L,int &R)
{
if(!u){L=R=0;return;}
pushdown(u);
if(size[son[u][0]]<k)
{
split(son[u][1],k-size[son[u][0]]-1,L,R);
son[u][1]=L,L=u,upt(u);
}
else
{
split(son[u][0],k,L,R);
son[u][0]=R,R=u,upt(u);
}
}
int build(int cnt)
{
static int stk[N],top=0;
for(int i=1;i<=cnt;i++)
{
int u=newnode(a[i]),pre=0;
while(top&&pri[u]<pri[stk[top]])
pre=stk[top--],upt(pre);
if(top)son[stk[top]][1]=u;
son[u][0]=pre,stk[++top]=u;
}
while(top)upt(stk[top--]);
return stk[1];
}
void erase(int x)
{
if(!x)return;
pool[++pool_top]=x;
erase(son[x][0]),erase(son[x][1]);
}
void Insert()
{
int pos=getint(),cnt=getint();
for(int i=1;i<=cnt;i++)a[i]=getint();
int x=build(cnt),L=0,R=0;
split(root,pos,L,R);
root=merge(merge(L,x),R);
}
void Delete()
{
int pos=getint(),cnt=getint();
int L=0,R=0,p=0,q=0;
split(root,pos-1,L,R),split(R,cnt,p,q);
erase(p);
root=merge(L,q);
}
void Make_same()
{
int pos=getint(),cnt=getint(),v=getint();
int L=0,R=0,p=0,q=0;
split(root,pos-1,L,R),split(R,cnt,p,q);
cover(p,v);
root=merge(merge(L,p),q);
}
void Reverse()
{
int pos=getint(),cnt=getint();
int L=0,R=0,p=0,q=0;
split(root,pos-1,L,R),split(R,cnt,p,q);
rever(p);
root=merge(merge(L,p),q);
}
void Get_sum()
{
int pos=getint(),cnt=getint();
int L=0,R=0,p=0,q=0;
split(root,pos-1,L,R);
split(R,cnt,p,q);
print(sum[p]),putchar('\n');
root=merge(merge(L,p),q);
}
int main()
{
//freopen("lx.in","r",stdin);
n=getint(),m=getint();mx[0]=-INF;
for(int i=1;i<=n;i++)a[i]=getint();
root=build(n);
char s[20];
while(m--)
{
scanf("%s",s);
if(s[0]=='I')Insert();
else if(s[0]=='D')Delete();
else if(s[0]=='R')Reverse();
else if(s[0]=='G')Get_sum();
else if(s[2]=='K')Make_same();
else print(mx[root]),putchar('\n');
}
return 0;
}