【codeforces 813E】【Army Creation】【数据结构】

题目大意

给出一列序列，不同的数代表不同的颜色。每次询问一段区间，求一个子集（每个数可选可不选）每种颜色的数不超过K，最大的集合大小是多少。

解题思路

先求出b[i],代表每个数往后的连续区间第一个同种颜色超过K个的位置是哪里。询问一个区间时所包含的b，b[i]在区间里面的数一定不可取。对于这个问题可以用主席树解决。

code

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LF double
#define LL long long
#define ULL unsigned long long
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define fr(i,j,k) for(int i=begin[j][k];i;i=next[j][i])
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;
}