Educational Codeforces Round 22 E. Army Creation 主席树

链接：

http://codeforces.com/contest/813/problem/E

题意：

给你一个长度问n的数组，每次讯问一段区间l，r，问最多可以选多少个数字使得每个数字出现的次数不超过k

题解：

使用主席树即可，具体看代码

代码：

struct Node { int l, r, sum; };
int n, k, q;
int a[MAXN], b[MAXN];
VI G[MAXN];
Node T[MAXN * 50];
int rt[MAXN], cnt;

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

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

int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    cin >> n >> k;
    rep(i, 1, n + 1) cin >> a[i], G[a[i]].pb(i);
    rep(i, 1, MAXN) if (G[i].size() > k) rep(j, k, G[i].size()) b[G[i][j]] = G[i][j - k];
    rep(i, 1, n + 1) {
        update(1, n, rt[i], rt[i - 1], i, 1);
        if (b[i]) update(1, n, rt[i], rt[i], b[i], -1);            
    }
    cin >> q;
    int last = 0;
    while (q--) {
        int l, r;
        cin >> l >> r;
        l = (l + last) % n + 1;
        r = (r + last) % n + 1;
        if (l > r) swap(l, r);
        last = query(1, n, rt[r], l, r);
        cout << last << endl;
    }
    return 0;
}