BZOJ3642 : [CEOI 2014] Cake

rank[i]表示第i美味的是哪块

left[i]表示在k左边美味度为i的是哪块

right[i]表示在k右边美味度为i的是哪块

用3棵线段树分别维护d序列的区间最大值、left序列的区间最大值、right序列的区间最小值

 

修改：

把第x块改成第y美味

把第y+1到第9美味的全部后移一位

然后把第x块美味度改成最大值+1

然后把第y-1到第1美味的美味度依次改成最大

 

查询：

设x到k这一段中美味度的最大值为y

求出k另一侧最靠近k的且美味度大于y的位置z

答案为|z-x|-1

 

时间复杂度为每次操作$O(\log n)$

 

一开始写了普通线段树被卡常数了TAT（最近怎么都被卡常数）

然后去学了zkw线段树后改成zkw线段树就过了，效果拔群，又快又短。

zkw线段树真是个好东西。

 

#include<cstdio>
const int N=250010,inf=~0U>>1;
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
inline int min(int a,int b){return a<b?a:b;}
inline int max(int a,int b){return a>b?a:b;}
inline void Max(int&a,int b){if(a<b)a=b;}
inline void Min(int&a,int b){if(a>b)a=b;}
int n,m,t,q,k,i,x,y,rank[N],d[N],left[5250010],right[5250010],M1,M2,ta_v[530000],tl_v[16777220],tr_v[16777220],que[500010][3];
char ch;
inline void ta_change(int x,int y){for(ta_v[x+=M1]=y,x>>=1;x;x>>=1)ta_v[x]=max(ta_v[x<<1],ta_v[x<<1|1]);}
inline int ta_ask(int x,int y){
  int t=-inf;
  for(x+=M1-1,y+=M1+1;x^y^1;x>>=1,y>>=1){
    if(~x&1)Max(t,ta_v[x^1]);
    if(y&1)Max(t,ta_v[y^1]);
  }
  return t;
}
inline void tl_change(int x,int y){for(tl_v[x+=M2]=y,x>>=1;x;x>>=1)tl_v[x]=max(tl_v[x<<1],tl_v[x<<1|1]);}
inline int tl_ask(int x,int y){
  int t=-inf;
  for(x+=M2-1,y+=M2+1;x^y^1;x>>=1,y>>=1){
    if(~x&1)Max(t,tl_v[x^1]);
    if(y&1)Max(t,tl_v[y^1]);
  }
  return t;
}
inline void tr_change(int x,int y){for(tr_v[x+=M2]=y,x>>=1;x;x>>=1)tr_v[x]=min(tr_v[x<<1],tr_v[x<<1|1]);}
inline int tr_ask(int x,int y){
  int t=inf;
  for(x+=M2-1,y+=M2+1;x^y^1;x>>=1,y>>=1){
    if(~x&1)Min(t,tr_v[x^1]);
    if(y&1)Min(t,tr_v[y^1]);
  }
  return t;
}
inline void modify(int x,int y){//d[x]改成y
  if(x<k)tl_change(d[x],0);
  if(x>k)tr_change(d[x],n+1);
  ta_change(x,d[x]=y);
  if(x<k)tl_change(y,x);
  if(x>k)tr_change(y,x);
}
inline void change(int x,int y){//把第x块蛋糕设置成第y美味
  int i,j=0;
  for(i=1;i<=10;i++)if(rank[i]==x)j=i;
  for(i=j?j:10;i>y;i--)rank[i]=rank[i-1];
  for(modify(rank[y]=x,++t),i=y-1;i;i--)modify(rank[i],++t);
}
inline int ask(int x){
  if(x==k)return 0;
  return x<k?tr_ask(ta_ask(x,k-1)+1,m)-x-1:x-tl_ask(ta_ask(k+1,x)+1,m)-1;
}
int main(){
  read(n),read(k);
  for(i=1;i<=n;i++)read(d[i]),rank[n-d[i]+1]=i;
  read(q);
  m=(t=n)+1;
  for(i=1;i<=q;i++){
    while(!(((ch=getchar())=='E')||(ch=='F')));
    if(ch=='E')read(que[i][1]),read(que[i][2]),m+=que[i][2];else que[i][0]=1,read(que[i][1]);
  }
  for(i=1;i<=m;i++)right[i]=n+1;
  for(i=1;i<k;i++)left[d[i]]=i;
  for(i=k+1;i<=n;i++)right[d[i]]=i;
  for(M1=1;M1<n+2;M1<<=1);
  for(i=1;i<=n;i++)ta_v[i+M1]=d[i];
  for(i=M1-1;i;i--)ta_v[i]=max(ta_v[i<<1],ta_v[i<<1|1]);
  for(M2=1;M2<m+2;M2<<=1);
  for(i=1;i<=m;i++)tl_v[i+M2]=left[i],tr_v[i+M2]=right[i];
  for(i=M2-1;i;i--)tl_v[i]=max(tl_v[i<<1],tl_v[i<<1|1]),tr_v[i]=min(tr_v[i<<1],tr_v[i<<1|1]);
  for(i=1;i<=q;i++)if(!que[i][0])change(que[i][1],que[i][2]);else printf("%d\n",ask(que[i][1]));
  return 0;
}