CF813E Army Creation (主席树)

题意:

n个数a[i]a[i],qq次询问,n,a[i],q<=10⁵
每次问[l,r]内最多可以选多少个数,满足同一个数的出现次数不超过k?

思路

假设我们想找一个 [l ,r ] 区间内的数,什么时候这个区间里的数是有效的呢?
那肯定是这个数 x 出现不超过 k 次,如果超过 k 次就无效了, 也就是说,向前找 和 这个数 x 一样的数,
直到向前找k 个,如果这个 数的位置 pos 小于 l, 那么这个数 x 一定是有效的.
而 x 的有效范围就是 pos 之后,也就是说 l 必须大于 pos ,这个 x 才可以有效
如果 x 的前面并没有 k 个相同的数,pos 就设为 0,
最后我们用主席树搞一下,统计一下 0, l-1 的值就可以了.

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+100;
vector<int>f[N];
int ls[N*40],rs[N*40],num[N*40],cnt,rt[N];
int n,m,a[N];
void modify(int &now, int l, int r, int k){
    ++cnt; ls[cnt] = ls[now], rs[cnt] = rs[now],num[cnt] = num[now] + 1; now = cnt;
    if (l + 1 == r) return;
    int mid = (l + r) >> 1;
    if (k < mid) modify(ls[now],l,mid,k);
    if (k >= mid) modify(rs[now],mid,r,k);
}
int Query(int rt1, int rt2, int l, int r, int k){
    if (r-1 <= k) return num[rt2] - num[rt1];
    int mid = (l + r) >> 1;
    int ans = 0;
    ans += Query(ls[rt1],ls[rt2],l,mid,k);
    if (k >= mid) ans += Query(rs[rt1],rs[rt2],mid,r,k);
    return ans;
}
int main(){
    int x,y,z;
    scanf("%d%d",&n,&m);
    for (int i = 1; i <= n; ++i){
        scanf("%d",&x);
        f[x].push_back(i);
        y = 0; z = f[x].size();
        if (z > m) y = f[x][z - m -1];
        rt[i] = rt[i-1];
        modify(rt[i],0,n+1,y);
    }
    scanf("%d",&m);
    int ans = 0;
    int l,r;
    for (int i = 0; i < m; ++i){
        scanf("%d%d",&x,&y);
        l = (x + ans) % n + 1;
        r = (y + ans) % n + 1; 
        if (l > r) swap(l,r);
        // printf("%d %d\n",l,r);
        ans = Query(rt[l-1],rt[r],0,n+1,l-1);
        printf("%d\n",ans);
    }
    return 0;
}