【分块】bzoj2120 数颜色

分块，在每个点记录一下它之前离它最近的相同颜色的位置pre[i]，显然问题转化成了求[l,r]中pre[i]<l的值的个数。

这是分块擅长的，在每个块内记录有序表，查询时对零散的暴力，整块的二分即可。

修改时有点麻烦，设修改a[p]。

可能会对pre[p]产生影响；

可能会对p位置之后的第一个 与a[p]修改前相等的值 的pre 产生影响；

可能会对p位置之后的第一个 与a[p]修改后相等的值 的pre 产生影响。

细节蛮多，调了很久。<---蒟蒻。

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,sz,sum,l[30],r[30],num[10001],pre[10001],preb[10001],a[10001],x,y,pos[1000001];
int Res,Num;char C,CH[12];
inline int G()
{
    Res=0;C='*'; 
    while(C<'0'||C>'9')C=getchar();
    while(C>='0'&&C<='9'){Res=Res*10+(C-'0');C=getchar();}
    return Res;
}
inline void P(int x)
{
    Num=0;if(!x){putchar('0');puts("");return;}
    while(x>0)CH[++Num]=x%10,x/=10;
    while(Num)putchar(CH[Num--]+48);
    puts("");
}
void makeblock()
{
    sz=sqrt((double)n*log2(n)); if(!sz) sz=1;
    for(sum=1;sum*sz<n;sum++)
      {
        l[sum]=(sum-1)*sz+1;
        r[sum]=sum*sz;
          for(int i=l[sum];i<=r[sum];i++)
          num[i]=sum;
      }
    l[sum]=sz*(sum-1)+1;
    r[sum]=n;
    for(int i=l[sum];i<=r[sum];i++)
      num[i]=sum;
}
void makepre()
{
    for(int i=1;i<=n;i++) {pre[i]=pos[a[i]]; pos[a[i]]=i;}
    memcpy(preb,pre,sizeof(pre));
    for(int i=1;i<=sum;i++) sort(pre+l[i],pre+r[i]+1);
}
inline void query()
{
    int res=0;
    if(num[x]+1>=num[y]) {for(int i=x;i<=y;i++) if(preb[i]<x) res++;}
    else
      {
          for(int i=x;i<=r[num[x]];i++) if(preb[i]<x) res++;
          for(int i=l[num[y]];i<=y;i++) if(preb[i]<x) res++;
          for(int i=num[x]+1;i<num[y];i++) res+=lower_bound(pre+l[i],pre+r[i]+1,x)-(pre+l[i]);
      }
    P(res);
}
inline void update()
{
    int t=0;
    for(int i=x+1;i<=n;i++)
      if(a[i]==y)
        {*lower_bound(pre+l[num[i]],pre+r[num[i]],preb[i])=x; preb[i]=x;
        sort(pre+l[num[i]],pre+r[num[i]]+1); break;}
    a[0]=y;
    for(int i=x-1;i>=0;i--)
      if(a[i]==y)
        {*lower_bound(pre+l[num[x]],pre+r[num[x]],preb[x])=i; preb[x]=i;
        sort(pre+l[num[x]],pre+r[num[x]]+1); break;}
    int t2=a[x]; a[x]=y;
    for(int i=x;i>=1;i--)
      if(a[i]==t2)
        {t=i; break;}
    for(int i=x+1;i<=n;i++)
      if(a[i]==t2)
        {*lower_bound(pre+l[num[i]],pre+r[num[i]],preb[i])=t; preb[i]=t;
        sort(pre+l[num[i]],pre+r[num[i]]+1); break;}
}char op[1];
int main()
{
    n=G();m=G();
    for(int i=1;i<=n;i++) a[i]=G();
    makeblock(); makepre();
    for(int i=1;i<=m;i++)
      {
        scanf("%s",op);x=G();y=G();
          if(op[0]=='Q') query();
          else update();
      }
    return 0;
}