COGS.1901.[模板][国家集训队2011]数颜色(带修改莫队)

题目链接 COGS
BZOJ2120
洛谷P1903

/*
Add和Subd函数中的vis不能直接设为=1或=0 
比如 l=1,r=0 -> l=3,r=5 时，[1,5]的vis标记全都是1 
所以要进行取反 
*/
#include<cmath>
#include<cstdio>
#include<cctype>
#include<algorithm>
using namespace std;
const int N=1e4+5,M1=1e4+5,M2=1005,MAX=1e6+5;

int n,m,size,A[N],Ans[M1],times[MAX],qn,mn,belong[N],now;
int H[N];
bool vis[N];
struct QUERY
{
    int l,r,id,t;//t:修改了几次后的 
    bool operator <(const QUERY &x)const
    {
        if(belong[l]==belong[x.l])
        {
            if(belong[r]==belong[x.r]) return t<x.t;
            return r<x.r;
        }
        return l<x.l;
    }
}q[M1];
struct MODIFT
{
    int pos,val,bef;
}optm[M2];

inline int read()
{
    int now=0,f=1;register char c=getchar();
    for(;!isdigit(c);c=getchar())
      if(c=='-') f=-1;
    for(;isdigit(c);now=now*10+c-'0',c=getchar());
    return now*f;
}

void Change(int t)
{
    int pos=optm[t].pos;
    if(vis[pos])//在询问区间内 
    {
        --times[A[pos]];
        if(!times[A[pos]]) --now;
    }
    optm[t].bef=A[pos],
    A[pos]=optm[t].val;//记录下改之前的值，然后更新 
    if(vis[pos])
    {
        if(!times[A[pos]]) ++now;
        ++times[A[pos]];
    }
}
void Unchange(int t)//改回来 
{
    int pos=optm[t].pos;
    if(vis[pos])
    {
        --times[A[pos]];
        if(!times[A[pos]]) --now;
    }
    A[pos]=optm[t].bef;
    if(vis[pos])
    {
        if(!times[A[pos]]) ++now;
        ++times[A[pos]];
    }
}
//void Change(int p,int v)//Change和Unchange可以由这一个函数代替 
//{
//  if(vis[p])
//      Calc(p), A[p]=v, Calc(p);
//  else
//      A[p]=v;
//}
void Add(int p)
{
    if(!times[A[p]]) ++now;
    ++times[A[p]];
    vis[p]^=1;
//  vis[p]=1;//加入访问区间 //错的！ 
}
void Subd(int p)
{
    --times[A[p]];
    if(!times[A[p]]) --now;
    vis[p]^=1;
//  vis[p]=0;//从区间中删掉 //错的！ 
}
void Calc(int p)
{
    if(vis[p])
    {
        if(--times[A[p]]==0)
            --now;
    }
    else if(++times[A[p]]==1)
        ++now;
    vis[p]^=1;
//  vis[p]=!vis[p];
}

int main()
{
    freopen("nt2011_color1.in","r",stdin);
    freopen("nt2011_color.out","w",stdout);

    n=read(),m=read();
    size=pow(n,2.0/3.0)*0.7;//取小一点更优 
    for(int i=1;i<=n;++i)
        A[i]=read(), H[i]=A[i], belong[i]=(i-1)/size+1;
    char opt[3];
    for(int a,b,i=1;i<=m;++i)
    {
        scanf("%s",opt);a=read(),b=read();
        if(opt[0]=='Q')
            q[++qn]=(QUERY){a,b,qn,mn};//注意编号 
        else
            optm[++mn].pos=a, optm[mn].val=b;//, optm[mn].bef=H[a], H[a]=b;
    }
    sort(q+1,q+1+qn);
    for(int i=1,l=1,r=0,t=0;i<=qn;++i)
    {
        int ln=q[i].l,rn=q[i].r,tn=q[i].t;
        while(t<tn) Change(++t);
        while(t>tn) Unchange(t--);
//      while(t<tn) ++t,Change(optm[t].pos,optm[t].val);//++t不能放在参数里写！ 
//      while(t>tn) Change(optm[t].pos,optm[t].bef),--t;
        
        while(l<ln) Calc(l++);
        while(l>ln) Calc(--l);
        while(r<rn) Calc(++r);
        while(r>rn) Calc(r--);
//      while(l<ln) Subd(l++);
//      while(l>ln) Add(--l);
//      while(r<rn) Add(++r);
//      while(r>rn) Subd(r--);
        Ans[q[i].id]=now;
    }
    for(int i=1;i<=qn;++i)
        printf("%d\n",Ans[i]);

    fclose(stdin);fclose(stdout);
    return 0;
}

转载于:https://www.cnblogs.com/SovietPower/p/8435098.html