Codeforces 455D Serega and Fun【解法二】

最新推荐文章于 2021-05-28 11:21:05 发布

原创最新推荐文章于 2021-05-28 11:21:05 发布 · 1.1k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

数据结构同时被 2 个专栏收录

148 篇文章

订阅专栏

codeforces

87 篇文章

订阅专栏

本文介绍了一种使用分块链表优化的数据结构方法，将序列分为若干块，每块使用链表维护，并记录每种颜色出现的次数。通过这种方法，可以在O(n√)的时间复杂度内完成元素的修改和查询操作。

把序列分成 $\sqrt n$ 块，每一块用链表维护元素，再维护每种颜色出现的次数。每次修改直接暴力移动并且修改颜色计数，这些都可以 $O(\sqrt n)$ 完成。询问的时候整块直接统计，不完整的暴力。也是 $O(\sqrt n)$ 的。注意到修改太多以后可能让块的大小不平衡，我们每 $\sqrt n$ 次修改之后 $O(n)$ 重构一次，这样保证任何时刻块的大小都是 $O(\sqrt n)$ 。

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=100010,maxm=320;
int a[maxn],val[maxn],cnt[maxm][maxn],edi[maxm][maxn],head[maxm],next[maxn],L[maxn],R[maxn],n,m,q,clo;
void init()
{
    clo++;
    for (int i=1;i<=m;i++)
    {
        L[i]=R[i-1]+1;
        R[i]=(i==m?n:L[i]+m-1);
        head[i]=a[L[i]];
        for (int j=L[i];j<R[i];j++) next[a[j]]=a[j+1];
        next[a[R[i]]]=0;
        for (int j=head[i];j;j=next[j])
            if (edi[i][val[j]]<clo)
            {
                edi[i][val[j]]=clo;
                cnt[i][val[j]]=1;
            }
            else cnt[i][val[j]]++;
    }
}
void modi(int l,int r)
{
    int pl,pr,u,v,x;
    for (int i=1;i<=m;i++)
        if (L[i]<=r&&r<=R[i])
        {
            pr=i;
            v=head[i];
            for (int j=L[i]+1;j<=r;j++)
            {
                x=v;
                v=next[v];
            }
            if (head[i]==v) head[i]=next[v];
            else next[x]=next[v];
            cnt[i][val[v]]--;
        }
    for (int i=1;i<=m;i++)
        if (L[i]<=l&&l<=R[i])
        {
            pl=i;
            u=head[i];
            for (int j=L[i]+1;j<=l;j++)
            {
                x=u;
                u=next[u];
            }
            if (head[i]==u) head[i]=v;
            else next[x]=v;
            next[v]=u;
            if (edi[i][val[v]]<clo)
            {
                edi[i][val[v]]=clo;
                cnt[i][val[v]]=1;
            }
            else cnt[i][val[v]]++;
        }
    if (pl<pr)
    {
        R[pl]++;
        L[pr]++;
        for (int i=pl+1;i<pr;i++)
            L[i]++,R[i]++;
    }
}
int qry(int l,int r,int k)
{
    int ans=0,pl,pr;
    for (int i=1;i<=m;i++)
    {
        if (L[i]<=l&&l<=R[i]) pl=i;
        if (L[i]<=r&&r<=R[i]) pr=i;
    }
    if (pl==pr)
    {
        for (int i=head[pl],j=L[pl];j<=r;i=next[i],j++)
            if (j>=l&&val[i]==k) ans++;
        return ans;
    }
    for (int i=head[pl],j=L[pl];j<=R[pl];i=next[i],j++)
        if (j>=l&&val[i]==k) ans++;
    for (int i=head[pr],j=L[pr];j<=r;i=next[i],j++)
        if (val[i]==k) ans++;
    for (int i=pl+1;i<pr;i++)
        if (edi[i][k]==clo) ans+=cnt[i][k];
    return ans;
}
int main()
{
    //freopen("f.in","r",stdin);
    int last=0,now=0,opt,l,r,k,t;
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&val[i]);
    m=sqrt(n);
    for (int i=1;i<=n;i++) a[i]=i;
    init();
    scanf("%d",&q);
    while (q--)
    {
        scanf("%d%d%d",&opt,&l,&r);
        l=(l+last-1)%n+1;
        r=(r+last-1)%n+1;
        if (l>r) swap(l,r);
        if (opt==1)
        {
            modi(l,r);
            if (++now==m)
            {
                t=0;
                for (int i=1;i<=m;i++)
                    for (int j=head[i];j;j=next[j])
                        a[++t]=j;
                init();
                now=0;
            }
        }
        else
        {
            scanf("%d",&k);
            k=(k+last-1)%n+1;
            printf("%d\n",last=qry(l,r,k));
        }
    }
}