主席树有关的一些题目（持续更新）

于 2021-02-09 20:45:06 发布
阅读量316 收藏 1
点赞数 1
分类专栏： 数据结构-主席树
版权声明：本文为博主原创文章，遵循 CC 4.0 BY 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_45483201/article/details/113775400
版权

主席树

模板

P3919 【模板】可持久化线段树 1（可持久化数组）

#include <bits/stdc++.h>

using namespace std;

const int N = 1e6 + 10;

int root[N], n, m;

int ls[N * 25], rs[N * 25], value[N * 25], tot;

void build(int &rt, int l, int r) {
  rt = ++tot;
  if (l == r) {
    scanf("%d", &value[rt]);
    return ;
  }
  int mid = l + r >> 1;
  build(ls[rt], l, mid);
  build(rs[rt], mid + 1, r);
}

void update(int &rt, int pre, int l, int r, int x, int v) {
  rt = ++tot, ls[rt] = ls[pre], rs[rt] = rs[pre], value[rt] = value[pre];
  if (l == r) {
    value[rt] = v;
    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(int rt, int l, int r, int x) {
  if (l == r) {
    return value[rt];
  }
  int mid = l + r >> 1;
  if (x <= mid) {
    return query(ls[rt], l, mid, x);
  }
  else {
    return query(rs[rt], mid + 1, r, x);
  }
}

int main() {
  // freopen("in.txt", "r", stdin);
  // freopen("out.txt", "w", stdout);
  // ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  scanf("%d %d", &n, &m);
  build(root[0], 1, n);
  for (int i = 1; i <= m; i++) {
    int pre, op, v, x;
    scanf("%d %d %d", &pre, &op, &x);
    if (op == 1) {
      scanf("%d", &v);
      update(root[i], root[pre], 1, n, x, v);
    }
    else {
      printf("%d\n", query(root[pre], 1, n, x));
      root[i] = root[pre];
    }
  }
  return 0;
}

P3834 【模板】可持久化线段树 2（主席树）

#include <bits/stdc++.h>

using namespace std;

const int N = 2e5 + 10;

int a[N], b[N], root[N], n, m;

int ls[N * 20], rs[N * 20], val[N * 20], tot;

void update(int &rt, int pre, int l, int r, int x, int value) {
  rt = ++tot, ls[rt] = ls[pre], rs[rt] = rs[pre], val[rt] = val[pre] + 1;
  if (l == r) {
    return ;
  }
  int mid = l + r >> 1;
  if (x <= mid) {
    update(ls[rt], ls[pre], l, mid, x, value);
  }
  else {
    update(rs[rt], rs[pre], mid + 1, r, x, value);
  }
}

int query(int rtl, int rtr, int l, int r, int k) {
  if (l == r) {
    return l;
  }
  int sum = val[ls[rtr]] - val[ls[rtl]], mid = l + r >> 1;
  if (sum >= k) {
    return query(ls[rtl], ls[rtr], l, mid, k);
  }
  else {
    return query(rs[rtl], rs[rtr], mid + 1, r, k - sum);
  }
}

int main() {
  // freopen("in.txt", "r", stdin);
  // freopen("out.txt", "w", stdout);
  // ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  scanf("%d %d", &n, &m);
  for (int i = 1; i <= n; i++) {
    scanf("%d", &a[i]);
    b[i] = a[i];
  }
  sort(b + 1, b + 1 + n);
  int n1 = unique(b + 1, b + 1 + n) - (b + 1);
  for (int i = 1; i <= n; i++) {
    int p = lower_bound(b + 1, b + 1 + n1, a[i]) - b;
    update(root[i], root[i - 1], 1, n1, p, 1);
  }
  for (int i = 1; i <= m; i++) {
    int l, r, k;
    scanf("%d %d %d", &l, &r, &k);
    printf("%d\n", b[query(root[l - 1], root[r], 1, n1, k)]);
  }
  return 0;
}

P2048 [NOI2010] 超级钢琴

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

struct Res {
  int value, id, k;

  Res(int _value = 0, int _id = 0, int _k = 0) : value(_value), id(_id), k(_k) {}

  bool operator < (const Res &t) const {
    return value < t.value;
  }
};

const int N = 5e5 + 10;

int a[N], b[N], root[N], n, k, L, R;

int ls[N << 6], rs[N << 6], val[N << 6], tot;

void update(int &rt, int pre, int l, int r, int x) {
  rt = ++tot, ls[rt] = ls[pre], rs[rt] = rs[pre], val[rt] = val[pre] + 1;
  if (l == r) {
    return ;
  }
  int mid = l + r >> 1;
  if (x <= mid) {
    update(ls[rt], ls[pre], l, mid, x);
  }
  else {
    update(rs[rt], rs[pre], mid + 1, r, x);
  }
}

int query(int rtl, int rtr, int l, int r, int k) {
  if (l == r) {
    return l;
  }
  int sum = val[ls[rtr]] - val[ls[rtl]], mid = l + r >> 1;
  if (sum >= k) {
    return query(ls[rtl], ls[rtr], l, mid, k);
  }
  else {
    return query(rs[rtl], rs[rtr], mid + 1, r, k - sum);
  }
}

priority_queue<Res> q;

int main() {
  // freopen("in.txt", "r", stdin);
  // freopen("out.txt", "w", stdout);
  // ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  scanf("%d %d %d %d", &n, &k, &L, &R);
  n += 1;
  for (int i = 2; i <= n; i++) {
    scanf("%d", &a[i]);
    a[i] += a[i - 1];
    b[i] = a[i];
  }
  sort(b + 1, b + 1 + n);
  int n1 = unique(b + 1, b + 1 + n) - (b + 1);
  for (int i = 1; i <= n; i++) {
    int p = lower_bound(b + 1, b + 1 + n1, a[i]) - b;
    update(root[i], root[i - 1], 1, n1, p);
  }
  for (int i = L + 1; i <= n; i++) {
    int l = max(1, i - R), r = i - L;
    int cur = a[i] - b[query(root[l - 1], root[r], 1, n1, 1)];
    q.push(Res(cur, i, 1));
  }
  ll ans = 0;
  for (int i = 1; i <= k; i++) {
    auto it = q.top();
    q.pop();
    ans += it.value;
    if (min(R - L + 1, it.id - L) == it.k) {
      continue;
    }
    int l = max(1, it.id - R), r = it.id - L;
    it.k++;
    int cur = a[it.id] - b[query(root[l - 1], root[r], 1, n1, it.k)];
    q.push(Res(cur, it.id, it.k));
  }
  printf("%lld\n", ans);
  return 0;
}

P2633 Count on a tree

主席树像是一种前缀和的数据结构，所以查询树上两点的第 k k k大，可以通过询问 s z [ u ] + s z [ v ] − s z [ l c a ( u , v ) ] − s z [ f a [ l c a ( u , v ) ] ] sz[u] + sz[v] - sz[lca(u, v)] - sz[fa[lca(u, v)]] sz[u]+sz[v]−sz[lca(u,v)]−sz[fa[lca(u,v)]]区间。

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5 + 10;

int head[N], to[N << 1], nex[N << 1], cnt = 1;

int value[N], a[N], root[N], n, m, n1;

int son[N], sz[N], fa[N], dep[N], top[N];

int ls[N * 20], rs[N * 20], sum[N * 20], num;

void add(int x, int y) {
  to[cnt] = y;
  nex[cnt] = head[x];
  head[x] = cnt++;
}

void update(int &rt, int pre, int l, int r, int x) {
  rt = ++num, ls[rt] = ls[pre], rs[rt] = rs[pre], sum[rt] = sum[pre] + 1;
  if (l == r) {
    return ;
  }
  int mid = l + r >> 1;
  if (x <= mid) {
    update(ls[rt], ls[pre], l, mid, x);
  }
  else {
    update(rs[rt], rs[pre], mid + 1, r, x);
  }
}

void dfs1(int rt, int f) {
  fa[rt] = f, dep[rt] = dep[f] + 1, sz[rt] = 1;
  int p = lower_bound(a + 1, a + 1 + n1, value[rt]) - a;
  update(root[rt], root[f], 1, n1, p);
  for (int i = head[rt]; i; i = nex[i]) {
    if (to[i] == f) {
      continue;
    }
    dfs1(to[i], rt);
    sz[rt] += sz[to[i]];
    if (!son[rt] || sz[son[rt]] < sz[to[i]]) {
      son[rt] = to[i];
    }
  }
}

void dfs2(int rt, int tp) {
  top[rt] = tp;
  if (!son[rt]) {
    return ;
  }
  dfs2(son[rt], tp);
  for (int i = head[rt]; i; i = nex[i]) {
    if (to[i] == fa[rt] || to[i] == son[rt]) {
      continue;
    }
    dfs2(to[i], to[i]);
  }
}

int lca(int x, int y) {
  while (top[x] != top[y]) {
    if (dep[top[x]] < dep[top[y]]) {
      swap(x, y);
    }
    x = fa[top[x]];
  }
  return dep[x] < dep[y] ? x : y;
}

int query(int x, int y, int f, int ff, int l, int r, int k) {
  if (l == r) {
    return l;
  }
  int ans = sum[ls[x]] + sum[ls[y]] - sum[ls[f]] - sum[ls[ff]], mid = l + r >> 1;
  if (ans >= k) {
    return query(ls[x], ls[y], ls[f], ls[ff], l, mid, k);
  }
  else {
    return query(rs[x], rs[y], rs[f], rs[ff], mid + 1, r, k - ans);
  }
}

int main() {
  // freopen("in.txt", "r", stdin);
  // freopen("out.txt", "w", stdout);
  // ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  scanf("%d %d", &n, &m);
  for (int i = 1; i <= n; i++) {
    scanf("%d", &value[i]);
    a[i] = value[i];
  }
  for (int i = 1; i < n; i++) {
    int x, y;
    scanf("%d %d", &x, &y);
    add(x, y);
    add(y, x);
  }
  sort(a + 1, a + 1 + n);
  n1 = unique(a + 1, a + 1 + n) - (a + 1);
  dfs1(1, 0);
  dfs2(1, 1);
  int last = 0;
  for (int i = 1; i <= m; i++) {
    int u, v, k;
    scanf("%d %d %d", &u, &v, &k);
    u ^= last;
    int f = lca(u, v), ff = fa[f];
    last = query(root[u], root[v], root[f], root[ff], 1, n1, k);
    last = a[last];
    printf("%d\n", last);
  }
  return 0;
}

#1901. Zju2112 Dynamic Rankings

#include <bits/stdc++.h>

using namespace std;

const int N = 1e4 + 10, maxn = 1000000000;

int a[N], n, m;

int root[N], ls[N << 8], rs[N << 8], sum[N << 8], num;

int A[N], B[N], cnt1, cnt2;

inline int lowbit(int x) {
  return x & (-x);
}

void update(int &rt, int l, int r, int x, int v) {
  if (!rt) {
    rt = ++num;
  }
  sum[rt] += v;
  if (l == r) {
    return ;
  }
  int mid = l + r >> 1;
  if (x <= mid) {
    update(ls[rt], l, mid, x, v);
  }
  else {
    update(rs[rt], mid + 1, r, x, v);
  }
}

int query(int l, int r, int k) {
  if (l == r) {
    return l;
  }
  int cur = 0, mid = l + r >> 1;
  for (int i = 1; i <= cnt1; i++) {
    cur -= sum[ls[A[i]]];
  }
  for (int i = 1; i <= cnt2; i++) {
    cur += sum[ls[B[i]]];
  }
  if (cur >= k) {
    for (int i = 1; i <= cnt1; i++) {
      A[i] = ls[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = ls[B[i]];
    }
    return query(l, mid, k);
  }
  else {
    for (int i = 1; i <= cnt1; i++) {
      A[i] = rs[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = rs[B[i]];
    }
    return query(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);
  scanf("%d %d", &n, &m);
  for (int i = 1; i <= n; i++) {
    scanf("%d", &a[i]);
    for (int j = i; j <= n; j += lowbit(j)) {
      update(root[j], 0, maxn, a[i], 1);
    }
  }
  for (int i = 1; i <= m; i++) {
    char op[2];
    int u, v, k;
    scanf("%s %d %d", op, &u, &v);
    if (op[0] == 'Q') {
      scanf("%d", &k);
      cnt1 = cnt2 = 0;
      for (int i = u - 1; i; i -= lowbit(i)) {
        A[++cnt1] = root[i];
      }
      for (int i = v; i; i -= lowbit(i)) {
        B[++cnt2] = root[i];
      }
      printf("%d\n", query(0, maxn, k));
    }
    else {
      for (int i = u; i <= n; i += lowbit(i)) {
        update(root[i], 0, maxn, a[u], -1);
      }
      a[u] = v;
      for (int i = u; i <= n; i += lowbit(i)) {
        update(root[i], 0, maxn, a[u], 1);
      }
    }
  }
  return 0;
}

P4175 [CTSC2008]网络管理

#include <bits/stdc++.h>

using namespace std;

const int N = 8e4 + 10, maxn = 100000000;

int a[N], n, m;

int head[N], to[N << 1], nex[N << 1], cnt = 1;

int son[N], sz[N], l[N], r[N], dep[N], top[N], fa[N], tot;

int root[N], sum[N << 8], ls[N << 8], rs[N << 8], num;

int A[N], B[N], cnt1, cnt2;

inline int lowbit(int x) {
  return x & (-x);
}

inline void add(int x, int y) {
  to[cnt] = y;
  nex[cnt] = head[x];
  head[x] = cnt++;
}

void update(int &rt, int l, int r, int x, int v) {
  if (!rt) {
    rt = ++num;
  }
  sum[rt] += v;
  if (l == r) {
    return ;
  }
  int mid = l + r >> 1;
  if (x <= mid) {
    update(ls[rt], l, mid, x, v);
  }
  else {
    update(rs[rt], mid + 1, r, x, v);
  }
}

void dfs1(int rt, int f) {
  fa[rt] = f, l[rt] = ++tot, sz[rt] = 1, dep[rt] = dep[f] + 1;
  for (int i = l[rt]; i <= n; i += lowbit(i)) {
    update(root[i], 0, maxn, a[rt], 1);
  }
  for (int i = head[rt]; i; i = nex[i]) {
    if (to[i] == f) {
      continue;
    }
    dfs1(to[i], rt);
    sz[rt] += sz[to[i]];
    if (!son[rt] || sz[son[rt]] < sz[to[i]]) {
      son[rt] = to[i];
    }
  }
  r[rt] = tot;
  for (int i = r[rt] + 1; i <= n; i += lowbit(i)) {
    update(root[i], 0, maxn, a[rt], -1);
  }
}

void dfs2(int rt, int tp) {
  top[rt] = tp;
  if (!son[rt]) {
    return ;
  }
  dfs2(son[rt], tp);
  for (int i = head[rt]; i; i = nex[i]) {
    if (to[i] == fa[rt] || to[i] == son[rt]) {
      continue;
    }
    dfs2(to[i], to[i]);
  }
}

int lca(int x, int y) {
  while (top[x] != top[y]) {
    if (dep[top[x]] < dep[top[y]]) {
      swap(x, y);
    }
    x = fa[top[x]];
  }
  return dep[x] < dep[y] ? x : y;
}

int query(int l, int r, int k) {
  if (l == r) {
    return l;
  }
  int cur = 0, mid = l + r >> 1;
  for (int i = 1; i <= cnt1; i++) {
    cur += sum[ls[A[i]]];
  }
  for (int i = 1; i <= cnt2; i++) {
    cur -= sum[ls[B[i]]];
  }
  if (cur >= k) {
    for (int i = 1; i <= cnt1; i++) {
      A[i] = ls[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = ls[B[i]];
    }
    return query(l, mid, k);
  }
  else {
    for (int i = 1; i <= cnt1; i++) {
      A[i] = rs[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = rs[B[i]];
    }
    return query(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);
  scanf("%d %d", &n, &m);
  for (int i = 1; i <= n; i++) {
    scanf("%d", &a[i]);
  }
  for (int i = 1; i < n; i++) {
    int x, y;
    scanf("%d %d", &x, &y);
    add(x, y);
    add(y, x);
  }
  dfs1(1, 0);
  dfs2(1, 1);
  for (int i = 1; i <= m; i++) {
    int k, u, v;
    scanf("%d %d %d", &k, &u, &v);
    if (!k) {
      for (int i = l[u]; i <= n; i += lowbit(i)) {
        update(root[i], 0, maxn, a[u], -1);
      }
      for (int i = r[u] + 1; i <= n; i += lowbit(i)) {
        update(root[i], 0, maxn, a[u], 1);
      }
      a[u] = v;
      for (int i = l[u]; i <= n; i += lowbit(i)) {
        update(root[i], 0, maxn, a[u], 1);
      }
      for (int i = r[u] + 1; i <= n; i += lowbit(i)) {
        update(root[i], 0, maxn, a[u], -1);
      }
    }
    else {
      int f = lca(u, v), ff = fa[f], total = dep[u] + dep[v] - dep[f] - dep[ff];
      if (total < k) {
        puts("invalid request!");
      }
      else {
        cnt1 = cnt2 = 0;
        for (int i = l[u]; i; i -= lowbit(i)) {
          A[++cnt1] = root[i];
        }
        for (int i = l[v]; i; i -= lowbit(i)) {
          A[++cnt1] = root[i];
        }
        for (int i = l[f]; i; i -= lowbit(i)) {
          B[++cnt2] = root[i];
        }
        for (int i = l[ff]; i; i -= lowbit(i)) {
          B[++cnt2] = root[i];
        }
        k = total - k + 1;
        printf("%d\n", query(0, maxn, k));
      }
    }
  }
  return 0;
}

#279. [SYZOI Round1] 滑稽♂树

#include <bits/stdc++.h>

using namespace std;

const int N = 3e4 + 10, maxn = 10000;

int head[N], to[N << 1], nex[N << 1], cnt = 1;

int l[N], r[N], tot;

int a[N], n, m;

int root[N], ls[N << 8], rs[N << 8], sum[N << 8], num;

int A[N], B[N], cnt1, cnt2;

inline void add(int x, int y) {
  to[cnt] = y;
  nex[cnt] = head[x];
  head[x] = cnt++;
}

inline int lowbit(int x) {
  return x & (-x);
}

void update(int &rt, int l, int r, int x, int v) {
  if (!rt) {
    rt = ++num;
  }
  sum[rt] += v;
  if (l == r) {
    return; 
  }
  int mid = l + r >> 1;
  if (x <= mid) {
    update(ls[rt], l, mid, x, v);
  }
  else {
    update(rs[rt], mid + 1, r, x, v);
  }
}

void dfs(int rt, int fa) {
  l[rt] = ++tot;
  for (int i = l[rt]; i <= n; i += lowbit(i)) {
    update(root[i], 1, maxn, a[rt], 1);
  }
  for (int i = head[rt]; i; i = nex[i]) {
    if (to[i] == fa) {
      continue;
    }
    dfs(to[i], rt);
  }
  r[rt] = tot;
}

int query1(int l, int r, int k) {
  if (l == r) {
    return l;
  }
  int cur = 0, mid = l + r >> 1;
  for (int i = 1; i <= cnt1; i++) {
    cur -= sum[ls[A[i]]];
  }
  for (int i = 1; i <= cnt2; i++) {
    cur += sum[ls[B[i]]];
  }
  if (cur >= k) {
    for (int i = 1; i <= cnt1; i++) {
      A[i] = ls[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = ls[B[i]];
    }
    return query1(l, mid, k);
  }
  else {
    for (int i = 1; i <= cnt1; i++) {
      A[i] = rs[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = rs[B[i]];
    }
    return query1(mid + 1, r, k - cur);
  }
}

int query2(int l, int r, int L, int R) {
  if (l >= L && r <= R) {
    int res = 0;
    for (int i = 1; i <= cnt1; i++) {
      res -= sum[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      res += sum[B[i]];
    }
    return res;
  }
  int mid = l + r >> 1, res = 0;
  int A1[100], B1[100];
  if (L <= mid) {
    for (int i = 1; i <= cnt1; i++) {
      A1[i] = A[i];
      A[i] = ls[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B1[i] = B[i];
      B[i] = ls[B[i]];
    }
    res += query2(l, mid, L, R);
    for (int i = 1; i <= cnt1; i++) {
      A[i] = A1[i];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = B1[i];
    }
  }
  if (R > mid) {
    for (int i = 1; i <= cnt1; i++) {
      A1[i] = A[i];
      A[i] = rs[A[i]];
    }
    for (int i = 1; i <= cnt2; i++) {
      B1[i] = B[i];
      B[i] = rs[B[i]];
    }
    res += query2(mid + 1, r, L, R);
    for (int i = 1; i <= cnt1; i++) {
      A[i] = A1[i];
    }
    for (int i = 1; i <= cnt2; i++) {
      B[i] = B1[i];
    }
  }
  return res;
}

int main() {
  // freopen("in.txt", "r", stdin);
  // freopen("out.txt", "w", stdout);
  // ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  scanf("%d", &n);
  for (int i = 1; i <= n; i++) {
    scanf("%d", &a[i]);
  }
  for (int i = 1; i < n; i++) {
    int x, y;
    scanf("%d %d", &x, &y);
    add(x, y);
    add(y, x);
  }
  dfs(1, 0);
  scanf("%d", &m);
  for (int i = 1; i <= m; i++) {
    int op, u, x, y;
    scanf("%d %d %d", &op, &u, &x);
    if (op == 1) {
      cnt1 = cnt2 = 0;
      for (int i = l[u] - 1; i ; i -= lowbit(i)) {
        A[++cnt1] = root[i];
      }
      for (int i = r[u]; i; i -= lowbit(i)) {
        B[++cnt2] = root[i];
      }
      printf("%d\n", query1(1, maxn, x));
    }
    else if (op == 2) {
      scanf("%d", &y);
      cnt1 = cnt2 = 0;
      for (int i = l[u] - 1; i ; i -= lowbit(i)) {
        A[++cnt1] = root[i];
      }
      for (int i = r[u]; i; i -= lowbit(i)) {
        B[++cnt2] = root[i];
      }
      printf("%d\n", query2(1, maxn, x, y));
    }
    else {
      for (int i = l[u]; i <= n; i += lowbit(i)) {
        update(root[i], 1, maxn, a[u], -1);
      }
      a[u] = x;
      for (int i = l[u]; i <= n; i += lowbit(i)) {
        update(root[i], 1, maxn, a[u], 1);
      }
    }
  }
  return 0;
}

Sequence II

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

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

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

然后再在第 l l l棵主席树中查找第 k k 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;
}
_lifehappy_
关注
  • 1
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
主席树 常见题目类型总结(未完待续)
HeartFireY的博客
09-17 435
???? | Powered By HeartFireY 文章目录一、 主席树维护区间第K大二、主席树维护线段树区间修改(标记永久化)三、求区间内不同的数字个数/求区间大于KKK的数字有多少四、求区间小于等于KKK的数字个数(二分查询)五、求区间Mex六、求区间内出现次数大于>=k次的最前数七、主席树+树上路径 一、 主席树维护区间第K大 ???? | Require：主席树 主席树最基础的应用，对于给定的序列建立权值数组，对于每个元素建立一棵主席树，维护[1,n][1,n][1,n]的数字个数(
[学习笔记]主席树经典例题
domy79360的博客
02-19 216
level 1 level 1-1 模板-查询k小值 模板，不再赘述。 #include<iostream> #incl...
主席树总结（题目合集）
常胜将军的博客
03-04 1692
1、HDU 2665 题意：无修改区间第k大 思路：主席树（离线算法）网上都有各种详细的解释了，就不多说了。。。主席树的核心思想包括前缀和、二分查找、空间重复利用、转化（区间表示在此范围内的数的个数，即权值线段树）。时间和空间复杂度为nlogn。 代码： #include<iostream> #include<algorithm> using namespace...
牛客(多校4)：Ancient Distance
Mr顺
07-20 483
#include <bits/stdc++.h> using namespace std; const int N = 2e5 + 10; int n, a[N], f[N]; vector<int> g[N]; int dfn[N], dfc; int d[N]; void dfs (int u) { dfn[++ dfc] = u; for (int & v : g[u]) { d[v] = d[u] + 1; ...
主席树题集及小结
代码笔记本
12-03 769
怕期末月完了就忘完了,蒟蒻总结一波。 先来模板题:求静态区间第k小 https://www.luogu.com.cn/problem/P3834 对于板子的理解: 板子的博客 #include<iostream> #include<vector> #include<queue> #include<cstring> #include<cmath> #include<map> #include<set> #inclu
主席树_主席树模板_
09-30
供信息学奥林匹克竞赛选手使用 主席树模板
权值线段树和主席树入门
09-25
权值线段树和主席树入门PPT，权值线段树，顾名思义就是记录权值的线段树，普通的线段树直接以坐标为l，r建树，而权值线段树是以大小来建树，树上寸的信息是该权值的数量，而通过建树时二分从小到大的性质，可以用这...
主席树.ppt
07-08
主席树.ppt
区间第k大 主席树代码
11-03
普通的主席树 区间第k大 代码 包含了简单的注释，和线段树很相似
学生会主席竞选面试题目.pdf
11-09
学生会主席竞选面试题目.pdf
主席树题目整理
Harris-H的博客
07-13 452
主席树题目整理 前话：由于太菜，所以开个坑整理有关主席树的题目，可能会咕咕咕。 1.P3834 【模板】可持久化线段树 2（主席树） 区间第kkk小，转权值主席树，具体见代码。 #include<bits/stdc++.h> using namespace std; typedef long long ll; const int N=2e6+5,M=N*21,inf=0x3f3f3f3f,mod=1e9+7; #define mst(a) memset(a,0,sizeof a) #define
【BZOJ 2653】middle 主席树好题推荐
pbihao的博客
12-25 507
谈谈这一道题的最强烈的感受在于，我的主席树入门是经典题目区间第k大树查询（当然，因为poj上不会强制在线，还用树套树和整体二分写过那道题），因此，对于主席树的理解仅仅是一个区间历史版本维护的理解，即类似于树套树的外层区间树内层全值树。但是忽略了主席树的真正伟大之处在于可持久化，而可持久化的也能是权值。 对于这一道题，由于我们不知道中位数是多少，因此考虑二分，而关键之处在于如何二分cheak，可以
主席树题目清单
qq_39898877的博客
11-14 171
codeforces 840D (2400) codeforces 813E (2300)
主席树模板总结及题集
jerans的博客
08-11 2689
介绍篇：http://blog.csdn.net/metalseed/article/details/8045038  -> 理解篇（配图比较生动）：http://blog.csdn.net/regina8023/article/details/41910615  -> 题目：poj2104 区间查询第k大元素，非递归写法，非递归不好进行回溯操作，递归更好写一些 #include #inc
主席树题目总结
cy41的博客
06-13 300
前置知识：基于值域的整体第k大（权值线段树） 写主席树必定需要注意空间的开销，一定要开够 若长度为n，修改为m次，则空间复杂度为n+m*logn。 可以在我的博客里翻一翻2333。 基础问题，求解区间第k大，或者区间小于等于k的个数： poj-2104 HDU-4417 下面两个前置知识LCA P2633 Count on a tree Distance on the tree ...
关于主席树的入门，讲解和题单
weixin_33752045的博客
08-01 101
主席树真是神仙操作啊……搞了好久才弄懂一点点QAQ 参考文章：https://www.cnblogs.com/zyf0163/p/4749042.html 　　　　　https://blog.csdn.net/creatorx/article/details/75446472 　　　　　https://blog.csdn.net/jerans/article/details/75807...
[主席树]HDOJ2665 && POJ2104 && POJ2761
aicheng4794的博客
07-16 135
主席树真是神奇的物种！ Orz一篇资料 题意：给n、m 　　 下面有n个数 (编号1到n) 　　　有m个询问，询问的是上面的数的编号在[l,r]之间第k小的数 n、m的范围都是$10^5$ 是主席树的入门题 借此来学习一下主席树 主席数利用函数式线段树来维护数列，一般用来解决区间第k大问题 空间时间的复杂度小于树套树(常数小) 划分树也可以解...
最详细的主席树(不修改,待修改) BZOJ 1901
Bartholomew_的博客
04-13 9214
By Bartholomew 前置知识: 1.树状数组 2.线段树 主席树 模板是干什么的,其实就是询问区间第k大 不支持修改: 复杂度O(nlogn)O(nlogn)O(nlog_n) 带修: 复杂度O(n∗(logn)2)O(n∗(logn)2)O(n * (log_n)^2) 分析: 请大家耐心看完下面一道题目的做法,因为跟主席树的关系很大 引题 ...
主席树模板题
No Program No Life
01-29 562
POJ2104 POJ2761 HDU2665 题意都是区间求第k小主席树插入是按数组序列从小到大后排名的插入。 所以之后查询才可以按左右子树的个数递归查询#include <cstdio> #include <iostream> #include <algorithm> using namespace std; const int maxn = 100000 + 1
我的舍友是主席 我可以合法谋取哪些利益
最新发布
03-29
作为你的舍友是主席，你可以合法谋取以下利益： 1. 更好的住宿条件：作为主席的舍友，你可能会得到更好的住宿条件或更大的房间。 2. 更好的社交机会：作为主席的舍友，你可能会得到更多的社交机会，比如参加主席的...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值