Sequence II
题目大意是有mmm次询问,每次询问一段区间[l,r][l, r][l,r],从左到右,如果这个数是在这个区间第一次出现,则记录下其下标,
我们会得到一个新的数组,要求这个数组的中位数是什么。
考虑使用主席树来写,我们倒序插入这些数的信息并记录下这个数上一次是出现在哪,对于每个点我们将其下标在主席树中+1+1+1,然后对于上一次出现的下标,在主席树中−1-1−1,所以我们直接在第lll棵主席树中查询区间[l,r][l, r][l,r]的权值即是这个区间的不同数字个数,在这个区间里,对于相同的数字,我们只会认定最右边的那个一个数字,所以重复的数字会消除。
然后再在第lll棵主席树中查找第kkk大即可得到答案了。
#include <bits/stdc++.h>using namespace std;const int N = 2e5 + 10;int a[N], pos[N], n, m, ans;int root[N], ls[N << 6], rs[N << 6], sum[N << 6], num;void update(int &rt, int pre, int l, int r, int x, int v) {rt = ++num, ls[rt] = ls[pre], rs[rt] = rs[pre], sum[rt] = sum[pre] + v;if (l == r) {return ;}int mid = l + r >> 1;if (x <= mid) {update(ls[rt], ls[pre], l, mid, x, v);}else {update(rs[rt], rs[pre], mid + 1, r, x, v);}
}int query_sum(int rt, int l, int r, int L, int R) {if (!rt) {return 0;}if (l >= L && r <= R) {return sum[rt];}int mid = l + r >> 1, ans = 0;if (L <= mid) {ans += query_sum(ls[rt], l, mid, L, R);}if (R > mid) {ans += query_sum(rs[rt], mid + 1, r, L, R);}return ans;
}int query_k(int rt, int l, int r, int k) {if (l == r) {return l;}int cur = sum[ls[rt]], mid = l + r >> 1;if (cur >= k) {return query_k(ls[rt], l, mid, k);}else {return query_k(rs[rt], mid + 1, r, k - cur);}
}int main() {// freopen("in.txt", "r", stdin);// freopen("out.txt", "w", stdout);// ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);int T;scanf("%d", &T);for (int cas = 1; cas <= T; cas++) {printf("Case #%d:", cas);memset(pos, 0, sizeof pos);for (int i = 0; i <= n; i++) {root[i] = 0;}for (int i = 0; i <= num; i++) {ls[i] = rs[i] = sum[i] = 0;}num = ans = 0;scanf("%d %d", &n, &m);for (int i = 1; i <= n; i++) {scanf("%d", &a[i]);}for (int i = n; i >= 1; i--) {update(root[i], root[i + 1], 1, n, i, 1);if (pos[a[i]]) {update(root[i], root[i], 1, n, pos[a[i]], -1);}pos[a[i]] = i;}for (int i = 1; i <= m; i++) {int l, r;scanf("%d %d", &l, &r);l = (l + ans) % n + 1, r = (r + ans) % n + 1;if (l > r) {swap(l, r);}int sum = query_sum(root[l], 1, n, l, r);ans = query_k(root[l], 1, n, sum + 1 >> 1);printf(" %d", ans);}puts("");}return 0;
}