Educational Codeforces Round 22 E. Army Creation 主席树

链接：

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

题意：

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

题解：

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

代码：

31 struct Node { int l, r, sum; };
32 int n, k, q;
33 int a[MAXN], b[MAXN];
34 VI G[MAXN];
35 Node T[MAXN * 50];
36 int rt[MAXN], cnt;
37 
38 void update(int l, int r, int &x, int y, int pos, int val) {
39     T[++cnt] = T[y], T[cnt].sum += val, x = cnt;
40     if (l == r) return;
41     int m = (l + r) >> 1;
42     if (pos <= m) update(l, m, T[x].l, T[y].l, pos, val);
43     else update(m + 1, r, T[x].r, T[y].r, pos, val);
44 }
45 
46 int query(int l, int r, int x, int L, int R) {
47     if (L <= l && r <= R) return T[x].sum;
48     int m = (l + r) >> 1;
49     int ret = 0;
50     if (L <= m) ret += query(l, m, T[x].l, L, R);
51     if (R > m) ret += query(m + 1, r, T[x].r, L, R);
52     return ret;
53 }
54 
55 int main() {
56     ios::sync_with_stdio(false), cin.tie(0);
57     cin >> n >> k;
58     rep(i, 1, n + 1) cin >> a[i], G[a[i]].pb(i);
59     rep(i, 1, MAXN) if (G[i].size() > k) rep(j, k, G[i].size()) b[G[i][j]] = G[i][j - k];
60     rep(i, 1, n + 1) {
61         update(1, n, rt[i], rt[i - 1], i, 1);
62         if (b[i]) update(1, n, rt[i], rt[i], b[i], -1);            
63     }
64     cin >> q;
65     int last = 0;
66     while (q--) {
67         int l, r;
68         cin >> l >> r;
69         l = (l + last) % n + 1;
70         r = (r + last) % n + 1;
71         if (l > r) swap(l, r);
72         last = query(1, n, rt[r], l, r);
73         cout << last << endl;
74     }
75     return 0;
76 }