【BZOJ 2453】【JZOJ 2491】维护队列

Description

你小时候玩过弹珠吗？
　　小朋友A有一些弹珠，A喜欢把它们排成队列，从左到右编号为1到N。为了整个队列鲜艳美观，小朋友想知道某一段连续弹珠中，不同颜色的弹珠有多少。当然，A有时候会依据个人喜好，替换队列中某个弹珠的颜色。但是A还没有学过编程，且觉得头脑风暴太浪费脑力了，所以向你来寻求帮助。

Solution

此题直上莫队算法，
分块之后直接搞，加个修改操作即可，
复杂度：$O(n^{\frac{5}{3}})$

Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
#define asp1(q) {if(!z[q])ans++;z[q]++;}
#define asj1(q) {z[q]--;if(!z[q])ans--;}
using namespace std;
const int N=1000500;
int read(int &n)
{
    char ch=' ';int q=0,w=1;
    for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
    if(ch=='-')w=-1,ch=getchar();
    for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;
}
int m1,m,n,ans;
int a[N],s[N];
int z[N];
struct qqww
{int x,y,z,i,la;}b[N],b1[N];
bool PX(qqww q,qqww w)
{
    int x1=(q.x-1)/m1+1,x2=(w.x-1)/m1+1;
    int y1=ceil(1.0*q.y/m1),y2=ceil(1.0*w.y/m1);
    return(x1<x2||(x1==x2&&y1<y2)||(x1==x2&&y1==y2&&q.z<w.z)||(x1==x2&&y1==y2&&q.z==w.z&&q.x<w.x));
}
int main()
{
    int q,w,e;
    read(n);read(m);
    m1=pow(n,1.0/3);
    m1*=m1; 
    fo(i,1,n)z[i]=read(a[i]);
    e=0;
    fo(i,1,m)
    {
        char ch=' ';while(ch!='Q'&&ch!='R')ch=getchar();
        read(q),read(w);
        if(ch=='R')b1[++e].x=q,b1[e].y=w;
            else b[i-e].x=q,b[i-e].y=w,b[i-e].z=e,b[i-e].i=i-e;
    }
    m-=e;
    sort(b+1,b+1+m,PX);
    fo(i,1,e)b1[i].la=z[b1[i].x],z[b1[i].x]=b1[i].y;
    memset(z,0,sizeof(z));
    int l=b[1].x,r=b[1].x-1;ans=0;
    fo(i,1,m)
    {
        fo(j,b[i-1].z+1,b[i].z)
        {
            if(l<=b1[j].x&&b1[j].x<=r)
            {
                asj1(a[b1[j].x]);
                asp1(b1[j].y);
            }
            a[b1[j].x]=b1[j].y;
        }
        fod(j,b[i-1].z,b[i].z+1)
        {
            if(l<=b1[j].x&&b1[j].x<=r)
            {
                asj1(a[b1[j].x]);
                asp1(b1[j].la);
            }
            a[b1[j].x]=b1[j].la;
        }
        fo(j,r+1,b[i].y)asp1(a[j]);
        fod(j,r,b[i].y+1)asj1(a[j]);
        fo(j,l,b[i].x-1)asj1(a[j]);
        fod(j,l-1,b[i].x)asp1(a[j]);
        s[b[i].i]=ans;
        l=b[i].x,r=b[i].y;
    }
    fo(i,1,m)printf("%d\n",s[i]);
    return 0;
}