Educational Codeforces Round 22 E. Army Creation(主席树)

题目链接：Educational Codeforces Round 22 E. Army Creation

题意：

给你n个数和一个数k，然后有q个询问.

每个询问 有一个区间[l,r]，问你这个区间内在满足每一种数不超过k的情况下，最大能选多少个数出来。

强制在线。

题解：

一看就要用到主席树，和主席数求区间内有多少不同的数的个数处理方法相同。

依次将每个数插入，当这个数出现的个数等于k了，就把最前面的那个数删掉。

然后询问就访问root[r]就行了。

第一次写完数据结构没有调试一遍过样例,一遍AC,这感觉真爽。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;
typedef long long ll;

const int N=1e5+7;

struct node{int l,r,sum;}T[N*40];
int root[N],a[N],hd[N],cnt,ct[N];
vector<int>Q[N];
int n,k,q,l,r;

void update(int l,int r,int &x,int y,int pos,int v)
{
    T[x=++cnt]=T[y],T[cnt].sum+=v;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(mid>=pos)update(l,mid,T[x].l,T[y].l,pos,v);
    else update(mid+1,r,T[x].r,T[y].r,pos,v);
}

int query(int l,int r,int L,int R,int rt)
{
    if(L<=l&&r<=R)return T[rt].sum;
    int mid=l+r>>1,an=0;
    if(L<=mid)an+=query(l,mid,L,R,T[rt].l);
    if(R>mid)an+=query(mid+1,r,L,R,T[rt].r);
    return an;
}

int main()
{
    scanf("%d%d",&n,&k);
    F(i,1,n)scanf("%d",a+i);
    F(i,1,n)
    {
        update(1,n,root[i],root[i-1],i,1);
        Q[a[i]].push_back(i);
        if(ct[a[i]]==k)
        {
            int x=Q[a[i]][hd[a[i]]++];
            update(1,n,root[i],root[i],x,-1);
        }else ct[a[i]]++;
    }
    int ans=0;
    scanf("%d",&q);
    while(q--)
    {
        scanf("%d%d",&l,&r);
        l=(l+ans)%n+1,r=(r+ans)%n+1;
        if(l>r)swap(l,r);
        printf("%d
",ans=query(1,n,l,r,root[r]));
    }
    return 0;
}