[BZOJ2120]数颜色（带修改莫队）

题目描述

传送门

题解

和BZOJ2453相同，在这里可以看到分块的做法
而这道题同时又是一道带修莫队裸题
带修莫队大体方法如下：
1、将修改询问离线并分开，记录每一个修改之前最近的一次询问的编号
2、分块之后将区间排序，关键字为：左端点块的编号、右端点块的编号、记录的最近一次修改的编号
3、在查询每一次询问之前，判断当前做过的修改是否恰好是这次询问需要的修改，如果不够将其修改，修改多了的话恢复回去，注意如果修改的位置在前一个询问的区间内要更新答案
4、转移询问和普通莫队相同

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 10005

int n,m,cq,cc,block,now,ans;
struct data{int x,y,t,id,ans;}q[N],ch[N];
int c[N],num[N],last[N],cnt[N*100];

int cmp(data a,data b)
{
    return num[a.x]<num[b.x]||(num[a.x]==num[b.x]&&num[a.y]<num[b.y])||(num[a.x]==num[b.x]&&num[a.y]==num[b.y]&&a.t<b.t);
}
void modui(int x,int y,int val)
{
    for (int i=x;i<=y;++i)
    {
        if (val==1)
        {
            if (!cnt[c[i]]) ++ans;
            ++cnt[c[i]];
        }
        else
        {
            --cnt[c[i]];
            if (!cnt[c[i]]) --ans;
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;++i) scanf("%d",&c[i]);
    block=sqrt(n);
    for (int i=1;i<=n;++i) num[i]=(i-1)/block+1;
    for (int i=1;i<=m;++i)
    {
        char opt=getchar();
        while (opt!='R'&&opt!='Q') opt=getchar();
        if (opt=='R')
        {
            ++cc;
            scanf("%d %d",&ch[cc].x,&ch[cc].y);
        }
        else
        {
            ++cq;q[cq].id=cq;
            scanf("%d %d",&q[cq].x,&q[cq].y);
            q[cq].t=cc;
        }
    }
    sort(q+1,q+cq+1,cmp);now=q[1].t;
    for (int i=1;i<=q[1].t;++i)
    {
        last[i]=c[ch[i].x];
        c[ch[i].x]=ch[i].y;
    }
    modui(q[1].x,q[1].y,1);q[q[1].id].ans=ans;
    for (int i=2;i<=cq;++i)
    {
        if (now<q[i].t)
            for (int j=now+1;j<=q[i].t;++j)
            {
                if (ch[j].x>=q[i-1].x&&ch[j].x<=q[i-1].y)
                {
                    --cnt[c[ch[j].x]];
                    if (!cnt[c[ch[j].x]]) --ans;
                    if (!cnt[ch[j].y]) ++ans;
                    ++cnt[ch[j].y];
                }
                last[j]=c[ch[j].x];
                c[ch[j].x]=ch[j].y;
            }
        else if (now>q[i].t)
            for (int j=now;j>q[i].t;--j)
            {
                if (ch[j].x>=q[i-1].x&&ch[j].x<=q[i-1].y)
                {
                    --cnt[c[ch[j].x]];
                    if (!cnt[c[ch[j].x]]) --ans;
                    if (!cnt[last[j]]) ++ans;
                    ++cnt[last[j]];
                }
                c[ch[j].x]=last[j];
            }
        now=q[i].t;
        if (q[i].x>q[i-1].x) modui(q[i-1].x,q[i].x-1,-1);
        else modui(q[i].x,q[i-1].x-1,1);
        if (q[i].y>q[i-1].y) modui(q[i-1].y+1,q[i].y,1);
        else modui(q[i].y+1,q[i-1].y,-1);
        q[q[i].id].ans=ans;
    }
    for (int i=1;i<=cq;++i) printf("%d\n",q[i].ans);
}