【CF813E】Army Creation（主席树）

【CF813E】Army Creation（主席树）

题面

CF
洛谷
翻译 by ppl 见洛谷

题解

考虑最多只会有(K)个相同的数
那么，也就是说，如果一个数会被选
那么，和它相等的数中，在它前面的第(K)的位置应该小于(l)
所以开个(vector)直接搞
然后按照前面的第(K)个数的位置插入主席树
查询(0..l-1)的值即可

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 111111
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Node
{
	int ls,rs;
	int v;
}t[MAX<<5];
int tot,rt[MAX];
void Modify(int &x,int ff,int l,int r,int p,int w)
{
	t[x=++tot]=t[ff];t[x].v+=w;
	if(l==r)return;
	int mid=(l+r)>>1;
	if(p<=mid)Modify(t[x].ls,t[ff].ls,l,mid,p,w);
	else Modify(t[x].rs,t[ff].rs,mid+1,r,p,w);
}
int Query(int r1,int r2,int l,int r,int L,int R)
{
	if(L<=l&&r<=R)return t[r1].v-t[r2].v;
	int mid=(l+r)>>1,ret=0;
	if(L<=mid)ret+=Query(t[r1].ls,t[r2].ls,l,mid,L,R);
	if(R>mid)ret+=Query(t[r1].rs,t[r2].rs,mid+1,r,L,R);
	return ret;
}
vector<int> S[MAX];
int lst[MAX],n,K,a[MAX];
int main()
{
	n=read();K=read();
	for(int i=1;i<=n;++i)a[i]=read();
	for(int i=1;i<=n;++i)
	{
		if(S[a[i]].size()>=K)lst[i]=S[a[i]][S[a[i]].size()-K];
		S[a[i]].push_back(i);
	}
	for(int i=1;i<=n;++i)Modify(rt[i],rt[i-1],0,n,lst[i],1);
	int Q=read(),ans=0;
	while(Q--)
	{
		int l=(read()+ans)%n+1,r=(read()+ans)%n+1;
		if(l>r)swap(l,r);
		printf("%d
",ans=Query(rt[r],rt[l-1],0,n,0,l-1));
	}
	return 0;
}