【BZOJ】1500: [NOI2005]维修数列
【算法】splay
【题解】数据结构
感谢Occult的模板>_<：HYSBZ 1500 维修数列
#include<cstdio> #include<cctype> #include<cstring> #include<queue> #include<algorithm> using namespace std; const int maxn=5e5+10,inf=0x3f3f3f3f; int n,m,root,a[maxn],t[maxn][2],f[maxn],A[maxn],L[maxn],R[maxn],M[maxn],sum[maxn],en[maxn],g[maxn],s[maxn]; //关系类：f父亲 t儿子 | 数值类：num值 L左起最大子段和 R右起最大子段和 M最大子段和 sum和 s结点数 | 标记类：en覆盖标记 g翻转标记 queue<int>q; int read() { char c;int s=0,t=1; while(!isdigit(c=getchar()))if(c=='-')t=-1; do{s=s*10+c-'0';}while(isdigit(c=getchar())); return s*t; } int Node(int fa,int num) { int sz=q.front();q.pop(); t[sz][0]=t[sz][1]=en[sz]=g[sz]=0; s[sz]=1;f[sz]=fa;A[sz]=L[sz]=R[sz]=M[sz]=sum[sz]=num; return sz; } void count(int x)//与线段树更新不同，记得加自身 { L[x]=max(L[t[x][0]],sum[t[x][0]]+A[x]+max(0,L[t[x][1]])); R[x]=max(R[t[x][1]],sum[t[x][1]]+A[x]+max(0,R[t[x][0]])); M[x]=max(0,R[t[x][0]])+A[x]+max(0,L[t[x][1]]); M[x]=max(M[x],max(M[t[x][0]],M[t[x][1]])); sum[x]=sum[t[x][0]]+A[x]+sum[t[x][1]]; s[x]=s[t[x][0]]+1+s[t[x][1]]; } void pushdown(int x) { if(g[x]) { g[t[x][0]]^=1;g[t[x][1]]^=1; swap(L[t[x][0]],R[t[x][0]]); swap(L[t[x][1]],R[t[x][1]]); swap(t[x][0],t[x][1]); g[x]=0; } if(en[x]) { if(t[x][0]) { en[t[x][0]]=1; A[t[x][0]]=A[x]; sum[t[x][0]]=A[x]*s[t[x][0]]; L[t[x][0]]=R[t[x][0]]=M[t[x][0]]=A[x]>0?sum[t[x][0]]:A[x]; } if(t[x][1]) { en[t[x][1]]=1; A[t[x][1]]=A[x]; sum[t[x][1]]=A[x]*s[t[x][1]]; L[t[x][1]]=R[t[x][1]]=M[t[x][1]]=A[x]>0?sum[t[x][1]]:A[x]; } en[x]=0; } } void rotate(int x) { int k=x==t[f[x]][1]; int y=f[x]; t[y][k]=t[x][!k];f[t[x][!k]]=y; if(f[y])t[f[y]][y==t[f[y]][1]]=x;f[x]=f[y];f[y]=x; t[x][!k]=y; L[x]=L[y];R[x]=R[y];M[x]=M[y];sum[x]=sum[y];s[x]=s[y]; count(y); } void splay(int x,int r) { for(int fa=f[r];f[x]!=fa;)//因为y被旋走了，所以要判断f[y] { if(f[f[x]]==fa){rotate(x);break;} int X=x==t[f[x]][1],Y=f[x]==t[f[f[x]]][1];//等号别打少…… if(X^Y)rotate(x),rotate(x); else rotate(f[x]),rotate(x); } } void find(int &x,int k) { for(int i=x;i;) { pushdown(i); if(k<=s[t[i][0]]){i=t[i][0];continue;} if(k==s[t[i][0]]+1){splay(i,x);x=i;break;} k-=s[t[i][0]]+1;i=t[i][1];//else不安全。。。 } } void build(int fa,int &x,int l,int r) { if(l>r)return; int mid=(l+r)>>1; x=Node(fa,a[mid]); build(x,t[x][0],l,mid-1); build(x,t[x][1],mid+1,r); count(x);// } void insert() { int l=read(),k=read(); for(int i=1;i<=k;i++)a[i]=read(); int r;build(0,r,1,k); find(root,l+1);find(t[root][1],1); t[t[root][1]][0]=r;f[r]=t[root][1]; count(t[root][1]);count(root);//记得count } void erase(int x) { if(!x)return; q.push(x); erase(t[x][0]); erase(t[x][1]); } void del() { int l=read(),k=read(); find(root,l);find(t[root][1],k+1); erase(t[t[root][1]][0]); t[t[root][1]][0]=0; count(t[root][1]);count(root);// } void cover() { int l=read(),k=read(),num=read(); find(root,l);find(t[root][1],k+1); int y=t[t[root][1]][0]; en[y]=1;A[y]=num; sum[y]=s[y]*A[y]; L[y]=R[y]=M[y]=A[y]>0?sum[y]:A[y]; count(t[root][1]);count(root);// } void round() { int l=read(),k=read(); find(root,l);find(t[root][1],k+1); int y=t[t[root][1]][0]; g[y]^=1; swap(L[y],R[y]); count(t[root][1]);count(root);// } void getsum() { int l=read(),k=read(); find(root,l);find(t[root][1],k+1); printf("%d\n",sum[t[t[root][1]][0]]); } void getM() { int sz=s[root]; find(root,1); find(t[root][1],sz-1); printf("%d\n",M[t[t[root][1]][0]]); } char str[20]; int main() { n=read();m=read(); a[0]=a[n+1]=root=0; for(int i=1;i<=maxn;i++)q.push(i); L[0]=R[0]=M[0]=-inf;//使空结点干扰判断，过程中需保证任何时刻都不应修改到该点的值。 for(int i=1;i<=n;i++)a[i]=read(); build(0,root,0,n+1); for(int i=1;i<=m;i++) { scanf("%s",str); if(str[2]=='S')insert(); if(str[2]=='L')del(); if(str[2]=='K')cover(); if(str[2]=='V')round(); if(str[2]=='T')getsum(); if(str[2]=='X')getM(); } return 0; }
update：现在已改用fhq-treap代替splay。
