BZOJ2883 : gss2加强版

首先离散化颜色

设pre[x]表示与x颜色相同的点上一次出现的位置，对于每种颜色开一个set维护

修改时需要修改x、x修改前的后继、x修改后的后继

询问[l,r]等价于询问[l,r]内pre[x]<l的点的权值和

线段树套Treap维护

 

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<set>
#define N 100010
using namespace std;
typedef long long ll;
int n,m,i,x,y,pre,nxt,a[N],b[N<<1],C,que[N][3],loc[N<<1];char ch;ll ans;
set<int>T[N<<1];set<int>::iterator j,k;
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 lower(int x){
  int l=1,r=C,t,mid;
  while(l<=r)if(b[mid=(l+r)>>1]<=x)l=(t=mid)+1;else r=mid-1;
  return t;
}
struct node{
  int val,p;ll v,sum;node*l,*r;
  node(){val=v=sum=p=0;l=r=NULL;}
  inline void up(){sum=v+l->sum+r->sum;}
}*blank=new(node),*root[N<<2];
inline void Rotatel(node*&x){node*y=x->r;x->r=y->l;x->up();y->l=x;y->up();x=y;}
inline void Rotater(node*&x){node*y=x->l;x->l=y->r;x->up();y->r=x;y->up();x=y;}
void Ins(node*&x,int p,int v){
  if(x==blank){
    x=new(node);x->val=p;x->l=x->r=blank;x->v=x->sum=v;x->p=std::rand();
    return;
  }
  x->sum+=v;
  if(p==x->val){x->v+=v;return;}
  if(p<x->val){
    Ins(x->l,p,v);
    if(x->l->p>x->p)Rotater(x);
  }else{
    Ins(x->r,p,v);
    if(x->r->p>x->p)Rotatel(x);
  }
}
ll Ask(node*x,int p){
  if(x==blank)return 0;
  if(p==x->val)return x->l->sum;
  if(p>x->val)return x->v+x->l->sum+Ask(x->r,p);
  return Ask(x->l,p);
}
void build(int x,int a,int b){
  root[x]=blank;
  if(a==b)return;
  int mid=(a+b)>>1;
  build(x<<1,a,mid),build(x<<1|1,mid+1,b);
}
inline void add(int c,int d,int e){
  int x=1,a=1,b=n,mid;
  while(a<b){
    Ins(root[x],d,e);mid=(a+b)>>1;x<<=1;
    if(c<=mid)b=mid;else a=mid+1,x|=1;
  }
  Ins(root[x],d,e);
}
void ask(int x,int a,int b,int c,int d,int e){
  if(c<=a&&b<=d){ans+=Ask(root[x],e);return;}
  int mid=(a+b)>>1;
  if(c<=mid)ask(x<<1,a,mid,c,d,e);
  if(d>mid)ask(x<<1|1,mid+1,b,c,d,e);
}
int main(){
  blank->l=blank->r=blank;
  for(read(n),i=1;i<=n;i++)read(a[i]),b[++C]=a[i];
  for(read(m),i=1;i<=m;i++){
    while(!(((ch=getchar())=='U')||(ch=='Q')));
    read(que[i][1]),read(que[i][2]);
    if(ch=='U')b[++C]=que[i][2];else que[i][0]=1;
  }
  for(sort(b+1,b+C+1),i=1;i<=C;i++)T[i].insert(0),T[i].insert(n+1);
  for(build(i=1,1,n);i<=n;i++){
    T[a[i]=lower(a[i])].insert(i);
    add(i,loc[a[i]],b[a[i]]);
    loc[a[i]]=i;
  }
  for(i=1;i<=m;i++)if(que[i][0])ans=0,ask(1,1,n,que[i][1],que[i][2],que[i][1]),printf("%lld\n",ans);
  else{
    x=que[i][1],y=lower(que[i][2]);
    k=j=T[a[x]].find(x);k--;pre=*k;k=j;k++;nxt=*k;
    add(x,pre,-b[a[x]]);
    if(nxt<=n)add(nxt,x,-b[a[x]]),add(nxt,pre,b[a[x]]);
    T[a[x]].erase(x);
    T[a[x]=y].insert(x);
    k=j=T[y].find(x);k--;pre=*k;k=j;k++;nxt=*k;
    add(x,pre,b[y]);
    if(nxt<=n)add(nxt,pre,-b[y]),add(nxt,x,b[y]);
  }
  return 0;
}