Sequence II (HDU 5919)(主席树)

Sequence II

题目大意是有$m$次询问，每次询问一段区间$[l,r]$，从左到右，如果这个数是在这个区间第一次出现，则记录下其下标，

我们会得到一个新的数组，要求这个数组的中位数是什么。

考虑使用主席树来写，我们倒序插入这些数的信息并记录下这个数上一次是出现在哪，对于每个点我们将其下标在主席树中$+1$，然后对于上一次出现的下标，在主席树中$-1$，所以我们直接在第$l$棵主席树中查询区间$[l,r]$的权值即是这个区间的不同数字个数，在这个区间里，对于相同的数字，我们只会认定最右边的那个一个数字，所以重复的数字会消除。

然后再在第$l$棵主席树中查找第$k$大即可得到答案了。

#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;
}