题目大意
给出一列序列,不同的数代表不同的颜色。每次询问一段区间,求一个子集(每个数可选可不选)每种颜色的数不超过K,最大的集合大小是多少。
解题思路
先求出b[i],代表每个数往后的连续区间第一个同种颜色超过K个的位置是哪里。询问一个区间时所包含的b,b[i]在区间里面的数一定不可取。对于这个问题可以用主席树解决。
code
using namespace std;
int const mn=1e5+9,inf=1e9+7;
int n,K,tim,a[mn],b[mn],cnt[mn],pre[mn],next[mn],size[mn*20],son[mn*20][2];
void insert(int p,int q,int l,int r,int v){
int md=(l+r)/2;
if(l==r){
size[q]=size[p]+1;
return;
}
if(v<=md){
son[q][1]=son[p][1];
if(!son[q][0])son[q][0]=++tim;
insert(son[p][0],son[q][0],l,md,v);
}else{
son[q][0]=son[p][0];
if(!son[q][1])son[q][1]=++tim;
insert(son[p][1],son[q][1],md+1,r,v);
}
size[q]=size[son[q][0]]+size[son[q][1]];
}
int qury(int p,int l,int r,int v){
int md=(l+r)/2;
if(l==r)return size[p];
if(v<=md)return qury(son[p][0],l,md,v);
else return size[son[p][0]]+qury(son[p][1],md+1,r,v);
}
int main(){
//freopen("d.in","r",stdin);
//freopen("d.out","w",stdout);
scanf("%d%d",&n,&K);
fo(i,1,n)scanf("%d",&a[i]);
int q;scanf("%d",&q);
int j=1;tim=n+1;
fd(i,n,1){
if(pre[a[i]])next[i]=pre[a[i]];
pre[a[i]]=i;
}
fo(i,1,n){
if(!cnt[a[i]])pre[a[i]]=i;
if(cnt[a[i]]<K)cnt[a[i]]++;
else{
b[pre[a[i]]]=i;
pre[a[i]]=next[pre[a[i]]];
}
}
fo(i,1,n){
if(!b[i])b[i]=n+1;
insert(i-1,i,1,n+1,b[i]);
}
int last=0;
fo(cas,1,q){
int l,r;
scanf("%d%d",&l,&r);
l=(l+last)%n+1;
r=(r+last)%n+1;
if(l>r)swap(l,r);
printf("%d\n",last=r-l+1-(qury(r,1,n+1,r)-qury(r,1,n+1,l-1)
-(qury(l-1,1,n+1,r)-qury(l-1,1,n+1,l-1))));
}
return 0;
}