bzoj2120 数颜色

Description

墨墨购买了一套N支彩色画笔（其中有些颜色可能相同），摆成一排，你需要回答墨墨的提问。墨墨会像你发布如下指令： 1、 Q L
R代表询问你从第L支画笔到第R支画笔中共有几种不同颜色的画笔。 2、 R P Col
把第P支画笔替换为颜色Col。为了满足墨墨的要求，你知道你需要干什么了吗？ Input

第1行两个整数N，M，分别代表初始画笔的数量以及墨墨会做的事情的个数。第2行N个整数，分别代表初始画笔排中第i支画笔的颜色。第3行到第2+M行，每行分别代表墨墨会做的一件事情，格式见题干部分。
Output

对于每一个Query的询问，你需要在对应的行中给出一个数字，代表第L支画笔到第R支画笔中共有几种不同颜色的画笔。

处理出每个位置的pre代表前一个和他相同的位置，那么询问实际就是询问在l..r中小于l的数的个数。分块以后排序，整块二分，边界暴力。
因为修改次数比较少，所以O(n+sqrt(n) * logn)修改可以接受。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int L[110],R[110],a[10010],pre[10010],f[10010],bel[10010],last[1000010],n,m,T;
void init()
{
    int i,j;
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++)
      scanf("%d",&a[i]);
    T=sqrt(n);
    for (i=1;i<=T;i++)
    {
        L[i]=R[i-1]+1;
        R[i]=i==T?n:L[i]+T-1;
        for (j=L[i];j<=R[i];j++)
          bel[j]=i;
    }
    for (i=1;i<=n;i++)
    {
        f[i]=pre[i]=last[a[i]];
        last[a[i]]=i;
    }
    for (i=1;i<=T;i++)
      sort(f+L[i],f+R[i]+1);
}
int qry(int l,int r)
{
    int i,j,ll,rr,mid,ans=0;
    for (i=1;i<=T;i++)
    {
        if (l<=L[i]&&R[i]<=r)
        {
            ll=L[i]-1;
            rr=R[i];
            while (ll<rr)
            {
                mid=(ll+rr+1)/2;
                if (f[mid]<l) ll=mid;
                else rr=mid-1;
            }
            ans+=ll-L[i]+1;
        }
        else
        {
            for (j=max(l,L[i]);j<=r&&j<=R[i];j++)
              if (pre[j]<l) ans++;
        }
    }
    return ans;
}
void modi(int p,int x)
{
    int i,j;
    a[p]=x;
    for (i=1;i<=n;i++)
      last[a[i]]=0;
    for (i=1;i<=n;i++)
    {
        if (pre[i]!=last[a[i]])
        {
            pre[i]=last[a[i]];
            for (j=L[bel[i]];j<=R[bel[i]];j++)
              f[j]=pre[j];
            sort(f+L[bel[i]],f+R[bel[i]]+1);
        }
        last[a[i]]=i;
    }
}
int main()
{
    int x,y;
    char s[3];
    init();
    while (m--)
    {
        scanf("%s%d%d",s,&x,&y);
        if (s[0]=='Q') printf("%d\n",qry(x,y));
        else modi(x,y);
    }
}