bzoj1500【NOI2005】维修数列

1500: [NOI2005]维修数列

Time Limit: 10 Sec   Memory Limit: 64 MB
Submit: 9542   Solved: 2907
[ Submit][ Status][ Discuss]

Description

Input

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

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

Source

Splay

题目意思很好理解，就是维护区间上各种各样的操作。方法也比较好想，Splay维护区间。

那么这道题的难点就在于代码实现了...考场上做这种题可真不是开玩笑的...我整整调了一下午加一晚上，做完之后整个人都不好了。

其实这道题应该多做一做，既可以享受码代码和调程序的乐趣，又可以对Splay的操作有更好地理解。（蒟蒻有发言权吗2333）

P.S. 由于自己在调程序时过于激动把U盘搞坏了 QAQ

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define LL long long
#define pa pair<int,int>
#define MAXN 1000005
#define INF 1000000000
using namespace std;
int n,m,rt,st,ed,tot1=0,tot2=0;
int fa[MAXN],t[MAXN][2],key[MAXN],a[MAXN],s[MAXN];
int sum[MAXN],mx[MAXN],lx[MAXN],rx[MAXN],cov[MAXN],size[MAXN];
bool rev[MAXN];
char op[20];
inline int read()
{
	int ret=0,flag=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') flag=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){ret=ret*10+ch-'0';ch=getchar();}
	return ret*flag;
}
inline void pushup(int k)
{
	if (!k) return;
	int l=t[k][0],r=t[k][1];
	sum[k]=sum[l]+sum[r]+key[k];
	lx[k]=max(lx[l],sum[l]+key[k]+max(0,lx[r]));
	rx[k]=max(rx[r],sum[r]+key[k]+max(0,rx[l]));
	mx[k]=max(max(0,rx[l])+key[k]+max(0,lx[r]),max(mx[l],mx[r]));
	size[k]=size[l]+size[r]+1;
}
inline void updatecov(int k,int x)
{
	if (!k) return;
	cov[k]=key[k]=x;
	sum[k]=x*size[k];
	mx[k]=lx[k]=rx[k]=max(x,sum[k]);
}
inline void updaterev(int k)
{
	if (!k) return;
	rev[k]^=1;
	swap(t[k][0],t[k][1]);
	swap(lx[k],rx[k]);
}
inline void pushdown(int k)
{
	if (!k) return;
	int l=t[k][0],r=t[k][1];
	if (rev[k])
	{
		updaterev(l);updaterev(r);
		rev[k]=0;
	}
	if (cov[k]!=-INF)
	{
		updatecov(l,cov[k]);updatecov(r,cov[k]);
		cov[k]=-INF;
	}
}
inline int newnode(int x)
{
	int tmp=tot2?s[tot2--]:++tot1;
	key[tmp]=a[x];
	cov[tmp]=-INF;
	rev[tmp]=0;
	lx[tmp]=rx[tmp]=mx[tmp]=-INF;
	return tmp;
}
inline int build(int l,int r)
{
	int mid=(l+r)>>1,lc=0,rc=0;
	if (l<mid) lc=build(l,mid-1);
	int tmp=newnode(mid);
	if (r>mid) rc=build(mid+1,r);
	if (lc) t[tmp][0]=lc,fa[lc]=tmp;
	if (rc) t[tmp][1]=rc,fa[rc]=tmp;
	pushup(tmp);
	return tmp;
}
inline void rotate(int x,int &k) 
{
	int y=fa[x],z=fa[y];
	int l=(t[y][1]==x),r=l^1;
	if (y==k) k=x;
	else if (t[z][0]==y) t[z][0]=x;else t[z][1]=x;
	fa[x]=z;fa[y]=x;fa[t[x][r]]=y;
	t[y][l]=t[x][r];t[x][r]=y;
	pushup(y);
}
inline void relax(int x,int k)
{
	if (x!=k) relax(fa[x],k);
	pushdown(x);
}
inline void splay(int x,int &k)
{
	relax(x,k);
	while (x!=k)
	{
		int y=fa[x],z=fa[y];
		if (y!=k)
		{
			if ((t[y][0]==x)^(t[z][0]==y)) rotate(x,k);
			else rotate(y,k);
		}
		rotate(x,k);
	}
	pushup(x);
}
inline int find(int k,int rank)
{
	pushdown(k);
	int l=t[k][0],r=t[k][1];
	if (size[l]+1==rank) return k;
	else if (size[l]>=rank) return find(l,rank);
	else return find(r,rank-size[l]-1);
}
inline void del(int &k)
{
	if (!k) return;
	s[++tot2]=k;
	fa[k]=0;
	del(t[k][0]);del(t[k][1]);
	lx[k]=rx[k]=mx[k]=-INF;
	k=0;
}
inline void solveins(int pos,int tot)
{
	F(i,1,tot) a[i]=read();
	int x=find(rt,pos+1),y=find(rt,pos+2);
	splay(x,rt);splay(y,t[x][1]);
	int tmp=build(1,tot);
	fa[tmp]=y;t[y][0]=tmp;
	pushup(y);pushup(x);
}
inline void solvedel(int pos,int tot)
{
	int x=find(rt,pos),y=find(rt,pos+tot+1);
	splay(x,rt);splay(y,t[x][1]);
	del(t[y][0]);
	pushup(y);pushup(x);
}
inline void solvecov(int pos,int tot,int z)
{
	int x=find(rt,pos),y=find(rt,pos+tot+1);
	splay(x,rt);splay(y,t[x][1]);
	updatecov(t[y][0],z);
	pushup(y);pushup(x);
}
inline void solverev(int pos,int tot)
{
	int x=find(rt,pos),y=find(rt,pos+tot+1);
	splay(x,rt);splay(y,t[x][1]);
	updaterev(t[y][0]);
	pushup(y);pushup(x);
}
inline int getsum(int pos,int tot)
{
	int x=find(rt,pos),y=find(rt,pos+tot+1);
	splay(x,rt);splay(y,t[x][1]);
	return sum[t[y][0]];
}
inline int getmax()
{
	splay(st,rt);splay(ed,t[rt][1]);
	return mx[t[ed][0]];
}
int main()
{
	n=read();m=read();
	F(i,2,n+1) a[i]=read();
	a[st=1]=0;a[ed=n+2]=0;
	rx[0]=lx[0]=mx[0]=-INF;
	rt=build(1,n+2);
	F(i,1,m)
	{
		scanf("%s",op);
		if (op[0]=='I'){int x=read(),y=read();solveins(x,y);}
		else if (op[0]=='D'){int x=read(),y=read();solvedel(x,y);}
		else if (op[2]=='K'){int x=read(),y=read(),z=read();solvecov(x,y,z);}
		else if (op[0]=='R'){int x=read(),y=read();solverev(x,y);}
		else if (op[0]=='G'){int x=read(),y=read();printf("%d\n",getsum(x,y));}
		else printf("%d\n",getmax());
	}
}