POJ 3580 splay

点击打开链接

题意：给个序列，然后有六种操作，分别对应一个功能

思路：splay的各种应用，两个标记，一个翻转的一个增加值的，模版参考kuangbin大神#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=200010;
int pre[maxn],ch[maxn][2],size[maxn],root,tot1;//父节点，左右孩子，子树规模，根节点，节点数量
int key[maxn],flag[maxn],rev[maxn],m[maxn];//该点的值，懒惰标记
int a[maxn],n,q;
int s[maxn],tot2;
//void treaval(int x){//debug部分
// if(x){
// treaval(ch[x][0]);
// printf("结点%2d:左儿子 %2d 右儿子 %2d 父结点 %2d size=%2d,key=%2d add=%2d\n",x,ch[x][0],ch[x][1],pre[x],size[x],key[x],flag[x]);
// treaval(ch[x][1]);
// }
//}
//void debug(){
// printf("root:%d\n",root);treaval(root);
//}
void newnode(int &r,int fa,int k){
if(tot2) r=s[tot2--];
else r=++tot1;
pre[r]=fa;size[r]=1;key[r]=m[r]=k;flag[r]=0;rev[r]=0;
ch[r][0]=ch[r][1]=0;
}
void update_rev(int r){
if(r==0) return ;
swap(ch[r][0],ch[r][1]);
rev[r]^=1;
}
void update_add(int r,int val){
if(r==0) return ;
flag[r]+=val;
key[r]+=val;
m[r]+=val;
}
void pushup(int r){
size[r]=size[ch[r][0]]+size[ch[r][1]]+1;
m[r]=key[r];
if(ch[r][0]) m[r]=min(m[r],m[ch[r][0]]);
if(ch[r][1]) m[r]=min(m[r],m[ch[r][1]]);
}
void pushdown(int r){
if(rev[r]){
update_rev(ch[r][0]);
update_rev(ch[r][1]);
rev[r]=0;
}
if(flag[r]){
update_add(ch[r][0],flag[r]);
update_add(ch[r][1],flag[r]);
flag[r]=0;
}
}
void buildtree(int &x,int l,int r,int fa){
if(l>r) return ;
int mid=(l+r)>>1;
newnode(x,fa,a[mid]);
buildtree(ch[x][0],l,mid-1,x);
buildtree(ch[x][1],mid+1,r,x);
pushup(x);
}
void init(){
root=tot1=tot2=0;
ch[root][0]=ch[root][1]=pre[root]=size[root]=flag[root]=0;
key[root]=0;m[root]=inf;
newnode(root,0,inf);
newnode(ch[root][1],root,inf);
buildtree(ch[ch[root][1]][0],1,n,ch[root][1]);
pushup(ch[root][1]);pushup(root);
}
void Rotate(int x,int kind){
int y=pre[x];
pushdown(y);pushdown(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;
pushup(y);
}
void splay(int r,int goal){
pushdown(r);
while(pre[r]!=goal){
if(pre[pre[r]]==goal){
pushdown(pre[r]);pushdown(r);
Rotate(r,ch[pre[r]][0]==r);
}else{
pushdown(pre[pre[r]]);pushdown(pre[r]);
pushdown(r);
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);
}
}
pushup(r);
if(goal==0) root=r;
}
int get_kth(int r,int k){
pushdown(r);
int t=size[ch[r][0]]+1;
if(t==k) return r;
if(t>k) return get_kth(ch[r][0],k);
else return get_kth(ch[r][1],k-t);
}
int get_min(int r){
pushdown(r);
while(ch[r][0]){
r=ch[r][0];
pushdown(r);
}
return r;
}
int get_max(int r){
pushdown(r);
while(ch[r][1]){
r=ch[r][1];
pushdown(r);
}
return r;
}
void add(int l,int r,int val){
splay(get_kth(root,l),0);
splay(get_kth(root,r+2),root);
update_add(ch[ch[root][1]][0],val);
pushup(ch[root][1]);
pushup(root);
}
void Reverse(int l,int r){
splay(get_kth(root,l),0);
splay(get_kth(root,r+2),root);
update_rev(ch[ch[root][1]][0]);
pushup(ch[root][1]);pushup(root);
}
void Revolve(int l,int r,int T){
int len=r-l+1;
T=(T%len+len)%len;
if(T==0) return ;
int c=r-T+1;
splay(get_kth(root,c),0);
splay(get_kth(root,r+2),root);
int tmp=ch[ch[root][1]][0];
ch[ch[root][1]][0]=0;
pushup(ch[root][1]);pushup(root);
splay(get_kth(root,l),0);
splay(get_kth(root,l+1),root);
ch[ch[root][1]][0]=tmp;
pre[ch[ch[root][1]][0]]=ch[root][1];
pushup(ch[root][1]);pushup(root);
}
void Insert(int x,int p){
splay(get_kth(root,x+1),0);
splay(get_kth(root,x+2),root);
newnode(ch[ch[root][1]][0],ch[root][1],p);
pushup(ch[root][1]);pushup(root);
}
void erase(int r){
if(r){
s[++tot2]=r;
erase(ch[r][0]);
erase(ch[r][1]);
}
}
void Delete(int x){
splay(get_kth(root,x),0);
splay(get_kth(root,x+2),root);
erase(ch[ch[root][1]][0]);pre[ch[ch[root][1]][0]]=0;
ch[ch[root][1]][0]=0;
pushup(ch[root][1]);pushup(root);
}
int query_min(int l,int r){
splay(get_kth(root,l),0);
splay(get_kth(root,r+2),root);
return m[ch[ch[root][1]][0]];
}
int main(){
char op[10];
int x,y,z;
while(scanf("%d",&n)!=-1){
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
init();
scanf("%d",&q);
while(q--){
scanf("%s",op);
if(op[0]=='A'){
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);
}else if(op[0]=='I'){
scanf("%d%d",&x,&y);
Insert(x,y);
}else if(op[0]=='D'){
scanf("%d",&x);
Delete(x);
}else if(op[0]=='M'){
scanf("%d%d",&x,&y);
printf("%d\n",query_min(x,y));
}else if(op[3]=='E'){
scanf("%d%d",&x,&y);
Reverse(x,y);
}else{
scanf("%d%d%d",&x,&y,&z);
Revolve(x,y,z);
}
}
}
return 0;
}