莫队算法（单点修改）bzoj2120数颜色

上一篇莫队算法（仅查询）：http://blog.csdn.net/winter2121/article/details/79051952

【个人理解】

带单点修改的莫队算法，需要多维护一个变量---时间。给每次修改操作标记先后时间T，当询问到区间[l,r]时，就把时间点调整到那个区间查询所处的时刻T。

分块排序时，将T作为最不优先的判定。即在初级莫队算法的基础上，排序时多考虑一个时间T。

每次区间查询时，将时间T调整到当前区间所处时间，再进行和初级莫队一样的查询操作。

【例题bzoj2120】http://www.lydsy.com/JudgeOnline/problem.php?id=2120

2120: 数颜色

Time Limit: 6 Sec   Memory Limit: 259 MB
Submit: 6860   Solved: 2758
[ Submit][ Status][ Discuss]

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支画笔中共有几种不同颜色的画笔。

Sample Input

6 5
1 2 3 4 5 5
Q 1 4
Q 2 6
R 1 2
Q 1 4
Q 2 6

Sample Output

4
4
3
4

HINT

对于100%的数据，N≤10000，M≤10000，修改操作不多于1000次，所有的输入数据中出现的所有整数均大于等于1且不超过10^6。

【代码】

#include<bits/stdc++.h>
using namespace std;
const int MAX=50100;
struct query{
    int l,r,T,id;//T时间,id询问
}q[MAX];
struct change{
    int pos,oldc,newc;//位置，旧状态，新状态
}c[MAX];//c[i]为时刻i的信息
int n,m,col[MAX],nowc[MAX];//col[i]为初始。nowc[i]为更新结点颜色的辅助数组
int unit,be[MAX];//unit块数，be[i]表示点i所属的块
int sum[MAX*100];//sum[i]表示区间[l,r]内颜色i的数目
int A[MAX];//A[i]为第i次询问的输出
bool cmp(query a,query b)
{
    if(be[a.l]==be[b.l])
        return be[a.r]==be[a.r] ? a.T<b.T : a.r<b.r;
    return be[a.l]<be[b.l];
}
void upans(int &ans,int color,int d)//更新ans
{//pos位置 d: 加入or离开
    sum[color]+=d;
    if(sum[color]==d)ans++;
    if(sum[color]==0)ans--;
}
void uppos(int &ans,int l,int r,change g,int d)//g更新点
{//d==1时正向更新，d==-1时恢复
    if(d<0)swap(g.oldc,g.newc);//时间回退，所以逆行
    if(l<=g.pos&&g.pos<=r){
        upans(ans,g.oldc,-1);//去旧
        upans(ans,g.newc,1);//来新
    }
    col[g.pos]=g.newc;
}
int main()
{
    scanf("%d%d",&n,&m);//点数，操作次数
    unit=pow(n,2.0/3);//块数
    for(int i=1;i<=n;i++){
        scanf("%d",&col[i]);
        nowc[i]=col[i];//动态变化
        be[i]=i/unit;//i所属块
    }
    int T=0,Q=0;//时间,询问数
    for(int i=0;i<m;i++)
    {
        int x,y;char ch[3];
        scanf("%s%d%d",ch,&x,&y);
        if(*ch=='Q')q[Q]=query{x,y,T,Q},Q++;//查询[x,y]
        if(*ch=='R'){c[++T]=change{x,nowc[x],y};nowc[x]=y;}//修改col[x]=y//T与新状态同步
    }
    memset(sum,0,sizeof(sum));
    sort(q,q+Q,cmp);//关键
    int ans=0,l=1,r=0;
    T=0;//时间恢复到初始0
    for(int i=0;i<Q;i++)//第i次询问
    {
        for(;T<q[i].T;T++)uppos(ans,l,r,c[T+1],1);
        for(;T>q[i].T;T--)uppos(ans,l,r,c[T],-1);

        for(;l<q[i].l;l++)upans(ans,col[l],-1);
        for(;r>q[i].r;r--)upans(ans,col[r],-1);
        for(;l>q[i].l;l--)upans(ans,col[l-1],1);
        for(;r<q[i].r;r++)upans(ans,col[r+1],1);
        A[q[i].id]=ans;
    }
    for(int i=0;i<Q;i++)
        printf("%d\n",A[i]);
}