树套树学习笔记 / 洛谷P2617 / bzoj1901【树套树】【主席树】【树状数组】【位运算】

树套树学习笔记

假设你已经学会了主席树……

树套树用来做的事情：

1. 查询区间 $k$ 大。
2. 查询区间 $k$ 小。
3. 修改一个数。
4. 修改一段区间。

主席树无法解决修改操作，因此前缀和修改一个数需要把后面的线段树都修改，所以将前缀和置换为树状数组，由于使用了两个数据结构，因此叫做树套树。

时间复杂度 $O(n{\log }^{2}n)$。

upd：主席树可以置换成值域线段树。

板子

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5 + 10;
struct Node {
  int l, r, sum;
} y[N * 400];
int n, m, cnt, cnt1, cnt2, z1[40], z2[40], root[N], a[N], k = 1e9;

int modify(int p, int l, int r, int i, int v) {
  int pp;
  if (p)
    pp = p;
  else
    pp = ++ cnt;
  y[pp] = y[p];
  if (l == r) {
    y[pp].sum += v;
    return pp;
  }
  int mid = l + r >> 1;
  if (i <= mid)
    y[pp].l = modify(y[p].l, l, mid, i, v);
  else
    y[pp].r = modify(y[p].r, mid + 1, r, i, v);
  y[pp].sum = y[y[pp].l].sum + y[y[pp].r].sum;
  return pp;
}

int query(int l, int r, int k) {
  if (l == r)
    return l;
  int mid = l + r >> 1;
  int t = 0;
  for (int i = 1; i <= cnt1; i ++)
    t -= y[y[z1[i]].l].sum;
  for (int i = 1; i <= cnt2; i ++)
    t += y[y[z2[i]].l].sum;
  if (t >= k) {
    for (int i = 1; i <= cnt2; i ++)
      z2[i] = y[z2[i]].l;
    for (int i = 1; i <= cnt1; i ++)
      z1[i] = y[z1[i]].l;
    return query(l, mid, k);
  } else {
    for (int i = 1; i <= cnt2; i ++)
      z2[i] = y[z2[i]].r;
    for (int i = 1; i <= cnt1; i ++)
      z1[i] = y[z1[i]].r;
    return query(mid + 1, r, k - t);
  }
}

void modify(int i, int p, int v) {
  for (; i <= n; i += (i & -i))
    root[i] = modify(root[i], 1, k, p, v);
}

signed main() {
  cin >> n >> m;
  for (int i = 1; i <= n; i ++)
    cin >> a[i];
  for (int i = 1; i <= n; i ++)
    modify(i, a[i], 1);
  while (m --) {
    char opt[2];
    cin >> opt;
    if (*opt == 'C') {
      int i, v;
      cin >> i >> v;
      modify(i, a[i], -1);
      a[i] = v;
      modify(i, a[i], 1);
    }
    else {
      int l, r, p;
      cin >> l >> r >> p;
      cnt1 = cnt2 = 0;
      for (int i = l - 1; i; i -= (i & -i))
        z1[++ cnt1] = root[i];
      for (int i = r; i; i -= (i & -i))
        z2[++ cnt2] = root[i];
      cout << query(1, k, p) << '\n';
    }
  }
}