[国家集训队] 数颜色

题目链接:戳我

带修莫队。时间复杂度\(O(N^{5/3})\)

不会的可以看看胡小兔 dalao的博客

先离线所有操作,然后标记上修改的时间戳,还按照普通莫队做,如果当前处理的询问在上一个处理的时间之前,就还原之间所有操作,如果在之后就先进行之间的所有操作。

排序按照块编号为第一关键字,右端点为第二关键字。

代码如下:

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define MAXN 1000010
using namespace std;
int n,m,cnt1,cnt2,ans,len;
int c[MAXN],cnt[MAXN],a[MAXN],ansans[MAXN];
struct Node{int l,r,id,t,ti;}q[MAXN];
struct Node2{int x,c,last;}node[MAXN];
inline bool cmp(struct Node x,struct Node y)
{
    if(x.t!=y.t) return x.t<y.t;
    return x.r<y.r;
}
inline void change(int pre,int now,int l,int r)
{
    if(pre==now) return;
    if(pre<now)
    {
        for(int i=pre+1;i<=now;i++)
        {
            if(node[i].x<l||node[i].x>r) c[node[i].x]=node[i].c;
            else
            {
                cnt[c[node[i].x]]--;
                if(cnt[c[node[i].x]]==0) ans--;
                cnt[node[i].c]++;
                if(cnt[node[i].c]==1) ans++;
                c[node[i].x]=node[i].c;
            }
        }
    }
    else
    {
        for(int i=pre;i>now;i--)
        {
            if(node[i].x<l||node[i].x>r) c[node[i].x]=node[i].last;
            else
            {
                cnt[c[node[i].x]]--;
                if(cnt[c[node[i].x]]==0) ans--;
                cnt[node[i].last]++;
                if(cnt[node[i].last]==1) ans++;
                c[node[i].x]=node[i].last;
            }
        }
    }
}
inline void count(int x,int k)
{
    cnt[c[x]]+=k;
    if(cnt[c[x]]==0&&k==-1) ans--;
    if(cnt[c[x]]==1&&k==1) ans++;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("ce.in","r",stdin);
    #endif
    scanf("%d%d",&n,&m);
    len=sqrt(n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),c[i]=a[i];
    for(int i=1;i<=m;i++)
    {
        char ch;int x,y;
        cin>>ch;
        scanf("%d%d",&x,&y);
        if(ch=='Q')
            cnt1++,q[cnt1]=(Node){x,y,cnt1,(x-1)/len+1,cnt2};
        else
        {
            cnt2++,node[cnt2]=(Node2){x,y,c[x]};
            c[x]=y;
        }
    }
    for(int i=1;i<=n;i++) c[i]=a[i];
    sort(&q[1],&q[cnt1+1],cmp);
    int l=1,r=0,pre=0;
    for(int i=1;i<=cnt1;i++)
    {
        if(pre!=q[i].ti) change(pre,q[i].ti,l,r);
        while(l<q[i].l) count(l++,-1);
        while(l>q[i].l) count(--l,+1);
        while(r<q[i].r) count(++r,+1);
        while(r>q[i].r) count(r--,-1);
        ansans[q[i].id]=ans;
        pre=q[i].ti;
    }
    for(int i=1;i<=cnt1;i++)
        printf("%d\n",ansans[i]);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/fengxunling/p/10310701.html
今日推荐