51nod 1471 小S的兴趣 分块

题意

小S喜欢有趣的事。但是,每个人的兴趣都是独特的。小S热衷于自问自答。有一天,小S想出了一个问题。
有一个包含n个正整数的数组a和针对这个数组的几个问题。这些问题有两种类型:
1. 在数组下标l到r的部分上,将一个单元格循环移动到右端。即以下面方式重新分配数组上的元素。
a[l], a[l+1], …, a[r-1], a[r] → a[r], a[l], a[l+1], …, a[r-1].
2. 在数组下标l到r的部分上,计算有多少元素的值与k相等。
小S很喜欢这个问题并且很快解决了它,你是否能够解决它呢?
1 ≤ n ≤ 10*5,1 ≤ q ≤ 10*5,强制在线

分析

由于这题的n,q比较大且时限只有1.5s,所以下意识地觉得分块不能过,于是就往各种数据结构方面想。。。
貌似是有一些奇奇怪怪的数据结构姿势可以过。
只要想到分块这题就不难了。
对每个块维护一个双向队列和桶,修改的话就头尾两块暴力搞,然后把前面每块的结尾扔到下一块的队头。
询问的话就中间的块扫桶两边暴力找即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

const int N=100005;
const int M=325;

int n,m,ls[N*2],nx[N*2],bel[N],sta[M],end[M],w[M][N],h[M],t[M],a[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void modify(int l,int r)
{
    if (l==r) return;
    if (bel[l]==bel[r])
    {
        int x=h[bel[l]],y=t[bel[l]];
        for (int i=sta[bel[l]];i<=l;i++) x=nx[x];
        for (int i=end[bel[l]];i>=r;i--) y=ls[y];
        ls[nx[y]]=ls[y];nx[ls[y]]=nx[y];
        nx[ls[x]]=y;ls[y]=ls[x];ls[x]=y;nx[y]=x;
        return;
    }
    int x=t[bel[l]],y=h[bel[r]];
    for (int i=end[bel[l]];i>=l;i--) x=ls[x];
    for (int i=sta[bel[r]];i<=r;i++) y=nx[y];
    w[bel[l]][a[y]]++;w[bel[r]][a[y]]--;
    ls[nx[y]]=ls[y];nx[ls[y]]=nx[y];
    nx[ls[x]]=y;ls[y]=ls[x];ls[x]=y;nx[y]=x;
    for (int i=bel[l];i<bel[r];i++)
    {
        int x=ls[t[i]],y=nx[h[i+1]];
        w[i][a[x]]--;w[i+1][a[x]]++;
        ls[nx[x]]=ls[x];nx[ls[x]]=nx[x];
        nx[ls[y]]=x;ls[x]=ls[y];ls[y]=x;nx[x]=y;
    }
}

int query(int l,int r,int k)
{
    int ans=0;
    if (bel[l]==bel[r])
    {
        int x=h[bel[l]],y=t[bel[l]];
        for (int i=sta[bel[l]];i<=l;i++) x=nx[x];
        for (int i=end[bel[l]];i>=r;i--) y=ls[y];
        while (x!=y) ans+=(a[x]==k),x=nx[x];ans+=(a[y]==k);
        return ans;
    }
    for (int i=bel[l]+1;i<bel[r];i++) ans+=w[i][k];
    int x=t[bel[l]],y=h[bel[r]];
    for (int i=end[bel[l]];i>=l;i--) x=ls[x],ans+=(a[x]==k);
    for (int i=sta[bel[r]];i<=r;i++) y=nx[y],ans+=(a[y]==k);
    return ans;
}

int main()
{
    n=read();int block=sqrt(n);
    for (int i=1;i<=n;i++)
    {
        a[i]=read();bel[i]=(i+block-1)/block;
        if (!sta[bel[i]]) sta[bel[i]]=i;
        end[bel[i]]=i;
    }
    for (int i=1;i<=bel[n];i++)
    {
        h[i]=n+i*2-1;t[i]=n+i*2;
        for (int j=sta[i];j<=end[i];j++) w[i][a[j]]++,nx[j]=j+1,ls[j]=j-1;
        ls[sta[i]]=h[i];nx[end[i]]=t[i];nx[h[i]]=sta[i];ls[t[i]]=end[i];
    }
    m=read();
    int ans=0;
    while (m--)
    {
        int op=read(),l=read(),r=read();
        l=(l+ans-1)%n+1;r=(r+ans-1)%n+1;
        if (l>r) swap(l,r);
        if (op==1) modify(l,r);
        else
        {
            int k=read();k=(k+ans-1)%n+1;
            printf("%d\n",ans=query(l,r,k));
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值