CodeForces 455D Serega and Fun (分块+双端队列)

题意:给出一个序列,两种操作
1.将区间[l,r]滚动一次
2.询问区间[l,r]中值等于k的数有多少个
在线

分块之后,两种操作都能在sqrt(n)的复杂度内解决,将每个块都设置为队列,这样滚动起来非常方便

一开始没有看到数值的范围,直接用的分块+bitset+双端队列,结果T了,后来看到了这一点,想写个分块+循环队列,结果发现编程复杂度太高..(好菜)…最终还是分块+双端队列…

#include<cstring>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>

using namespace std;

#define maxn 100005

int n,m;
int b[350][maxn];
deque<int>que[350];
int val[maxn];
int block,num,belong[maxn],l[maxn],r[maxn];

void init()
{
    block=sqrt(n);
    num=n/block;
    if(n&block) num++;
    for(int i=1; i<=n; i++)
    {
        belong[i]=(i-1)/block+1;
    }
    for(int i=1; i<=num; i++)
    {
        l[i]=(i-1)*block+1;
        r[i]=i*block;
    }
    r[num]=n;
    for(int i=1; i<=num; i++)
    {
        for(int j=l[i]; j<=r[i]; j++)
        {
            b[i][val[j]]++;
            que[i].push_back(val[j]);
        }
    }
}

void update(int x,int y)
{
    if(belong[x]==belong[y])
    {
        int index=belong[x];
        x-=l[index];
        y-=l[index];
        int temp=que[index][y];
        for(int i=y; i>x; i--)
        {
            que[index][i]=que[index][i-1];
        }
        que[index][x]=temp;
        return ;
    }
    int temp=que[belong[y]][y-l[belong[y]]];
    int temp2=que[belong[x]][r[belong[x]]-l[belong[x]]];
    b[belong[x]][temp2]--;
    b[belong[y]][temp]--;
    for(int i=r[belong[x]]; i>x; i--)
    {
        que[belong[x]][i-l[belong[x]]]=que[belong[x]][i-l[belong[x]]-1];
    }
    que[belong[x]][x-l[belong[x]]]=temp;
    b[belong[x]][temp]++;
    for(int i=belong[x]+1; i<belong[y]; i++)
    {
        int tp=temp2;
        b[i][temp2]++;
        temp2=que[i][r[i]-l[i]];
        que[i].pop_back();
        que[i].push_front(tp);
        b[i][temp2]--;
    }
    for(int i=y; i>l[belong[y]]; i--)
    {
        que[belong[y]][i-l[belong[y]]]=que[belong[y]][i-l[belong[y]]-1];
    }
    que[belong[y]][0]=temp2;
    b[belong[y]][temp2]++;
}

int query(int x,int y,int k)
{
    int ans=0;
    if(belong[x]==belong[y])
    {
        for(int i=x; i<=y; i++)
        {
            if(que[belong[x]][i-l[belong[x]]]==k) ans++;
        }
        return ans;
    }
    for(int i=x; i<=r[belong[x]]; i++)
    {
        if(que[belong[x]][i-l[belong[x]]]==k) ans++;
    }
    for(int i=belong[x]+1; i<belong[y]; i++)
    {
        ans+=b[i][k];
    }
    for(int i=l[belong[y]]; i<=y; i++)
    {
        if(que[belong[y]][i-l[belong[y]]]==k) ans++;
    }
    return ans;
}

int main()
{
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&val[i]);
    }
    init();
    scanf("%d",&m);
    int ans=0;
    while(m--)
    {
        int op;
        scanf("%d",&op);
        int x,y;
        scanf("%d%d",&x,&y);
        x=(x+ans-1)%n+1;
        y=(y+ans-1)%n+1;
        if(x>y) swap(x,y);
        if(op==1)
        {
            update(x,y);
        }
        else if(op==2)
        {
            int k;
            scanf("%d",&k);
            k=(k+ans-1)%n+1;
            printf("%d\n",ans=query(x,y,k));
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值