2016CCPC长春：Sequence II（主席树）

问题概述：n个数q次查询，每次查询(l,r)表示查询区间[l,r]内所有不同数字第一次出现的下标，输出这些下标中位数

特殊：如果下标数sum为偶数个，则输出第sum/2个下标，且强制在线，每一组的l和r都跟上一组的答案有关

（HDU 5919：http://acm.hdu.edu.cn/showproblem.php?pid=5919）

输入样例：                            对应输出：

1                                            Case #1: 3 1

5 2

2 5 2 1 2

2 3

2 4

解题思路：

将数组倒过来建树，每当添加一个数字时，设这个数字的下标为i，这个数字上次出现的下标为j，则下标i对应

的val值+1，下标j对应的val值-1，每次查询区间[l,r]直接询问版本为l的树即可，先找出在这个区间内的val值之和

sum，然后在这个区间内找到第(sum+1)/2个值为1的下标就是答案

#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;typedef struct{int l, r;int val;}Ctree;Ctree s[8000005];int n, tot, a[200005], t[200005], flag[200005], prt[200005];int Build(int l, int r);int Update(int root, int x, int k);int Query(int root, int l, int r, int a, int b);int Find(int root, int k);int main(void){int T, i, q, l, r, ans, sum, k, temp, cas = 1;scanf("%d", &T);while(T--){ans = tot = 0;memset(flag, -1, sizeof(flag));scanf("%d%d", &n, &q);for(i=1;i<=n;i++)scanf("%d", &a[i]);t[n+1] = Build(1, n);for(i=n;i>=1;i--){t[i] = Update(t[i+1], i, 1);if(flag[a[i]]!=-1)t[i] = Update(t[i], flag[a[i]], -1);/*直接另起一棵树，虽然有点浪费空间*/flag[a[i]] = i;}k = 0;while(q--){scanf("%d%d", &l, &r);temp = l;l = min((l+ans)%n+1, (r+ans)%n+1);r = max((temp+ans)%n+1, (r+ans)%n+1);sum = Query(t[l], 1, n, l, r);ans = Find(t[l], (sum+1)/2);prt[++k] = ans;}printf("Case #%d:", cas++);for(i=1;i<=k;i++)printf(" %d", prt[i]);printf("\n");}return 0;}int Build(int l, int r)/*建空树，所有节点val值为0*/{int m, root;m = (l+r)/2;root = ++tot;if(l==r){s[root].val = 0;return root;}s[root].l = Build(l, m);s[root].r = Build(m+1, r);s[root].val = s[s[root].l].val+s[s[root].r].val;return root;}int Update(int root, int x, int k)/*修改第x个下标的值*/{int now, temp, m, l, r;temp = now = ++tot;l = 1, r = n;while(l<r){m = (l+r)/2;s[now].val = s[root].val+k;if(x<=m){s[now].l = ++tot;s[now].r = s[root].r;root = s[root].l;r = m;now = tot;}else{s[now].l = s[root].l;s[now].r = ++tot;root = s[root].r;l = m+1;now = tot;}}s[now].val = s[root].val+k;return temp;}int Query(int root, int l, int r, int a, int b){int m, sum;if(l>=a && r<=b)return s[root].val;m = (l+r)/2;sum = 0;if(a<=m)sum += Query(s[root].l, l, m, a, b);if(b>=m+1)sum += Query(s[root].r, m+1, r, a, b);return sum;}int Find(int root, int k){int l, r, m;l = 1, r = n;while(l<r){m = (l+r)/2;if(s[s[root].l].val>=k){root = s[root].l;r = m;}else{k -= s[s[root].l].val;root = s[root].r;l = m+1;}}return l;}