CF 813E Army Creation --- 分块|主席树

一、分块

核心就是找到分块要维护的东西。想象一列数字中，我们取出其中相同的一组，我们要在给定范围内取不超过k个，如果暴力从左往右判断然后记录个数，很不现实吧。那我们就开个数组记录这个数在此组中的位置+k个后会不会超过这一组的个数，但之后统计的关键不是“从左往右”。如果碰到的数+k后还在区间内，我们不做记录，因为这样限制条件就会很多，分块询问时中间的块也不好问，很容易出错（但说不定可以写）。我们可以转换思路，专找+k后大于边界的，这说明我们拿上它个数也不会超过k，这样ans++；之后分块，两边一个一个找，中间用二分找到大于边界的，直接加，所以前面要sort预处理。

//写得乱七八糟的题解，可能只有我自己才看得懂~~

#include<bits/stdc++.h>
#include<iostream>
#include<stack>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#define mem(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define fio ios::sync_with_stdio(false);cin.tie(0)
#define ll long long
#define mp make_pair
#define inf 0x3f3f3f3f
const int N=1e5+5;
const int M=1e3+10;
const int mod=9973;
using namespace std;
int blo,num,belong[N],n,nx[N],l[N],a[N],b[N],r[N],k,mx=-inf;
vector<int>cnt[N];
void build()
{
    blo=sqrt(n);
    num=(n%blo)? n/blo+1:n/blo;
    for(int i=1;i<=n;i++)
    {
        belong[i]=(i-1)/blo+1;
        b[i]=nx[i];
    }
    for(int i=1;i<=num;i++)
    {
        l[i]=(i-1)*blo+1;
        r[i]=i*blo;
    }
    r[num]=n;
    for(int i=1;i<=num;i++)
        sort(b+l[i],b+r[i]+1);
}
inline int query(int x,int y)
{
    int ans=0;
    for(int i=x;i<=min(r[belong[x]],y);i++)
        if(nx[i]>y) ans++;
    if(belong[x]!=belong[y])
    {
        for(int i=l[belong[y]];i<=y;i++)
            if(nx[i]>y) ans++;
        for(int i=belong[x]+1;i<=belong[y]-1;i++)
        {
            int j=b+r[i]+1-upper_bound(b+l[i],b+r[i]+1,y);
            ans+=j;
        }
    }
    return ans;
}
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        cnt[a[i]].push_back(i);
        mx=max(mx,a[i]);
    }
    for(int i=1;i<=mx;i++)
    {
        for(int j=0;j<cnt[i].size();j++)
        {
           if(j+k>=cnt[i].size()) nx[cnt[i][j]]=N;
           else  nx[cnt[i][j]]=cnt[i][j+k];
        }
    }
    build();
    int m,x,y,last=0;
    cin>>m;
    while(m--)
    {
        cin>>x>>y;
        x=(x+last)%n+1;
        y=(y+last)%n+1;
        if(x>y) swap(x,y);
        int ans;
        if(y-x+1<=k) ans=y-x+1;
        else ans=query(x,y);
        cout<<ans<<endl;
        last=ans;
    }
    return 0;
}