主席树学习笔记（一）——静态

主席树学习笔记 —— By Menci

传统的线段树用于维护一条线段上的区间，可以方便地查询区间信息。而如果将线段树转化为『权值线段树』，每个叶子节点存储某个元素出现次数，一条线段的总和表示区间内所有数出现次数的总和。

但注意，权值线段树其实和区间线段树本质还是一样的。建树过程相同。
唯一区别就是叶子节点从存储权值变成了存储权值的出现次数。区间存储这段区间的所有权值出现的次数。

---

主席树（静态） 图文讲解 （这个比较适合基础理解静态主席树

主席树就是利用函数式编程的思想来使线段树支持询问历史版本、同时充分利用它们之间的共同数据来减少时间和空间消耗的增强版的线段树。

主席树就是很多线段树的总体，而这些线段树就是按给定序列的所有前缀建立的。

文中说用前缀和更新，其实就是用前缀和在空树基础上建树。

设原序列有 n 个数，含有 m 次询问
空间复杂度：（建空树）$4 * n$ + （前缀和建树）$nlog_2n$
一般我们数组大小就开 $nlog^2n$ （我反正就直接开 * 40 的
时间复杂度：$m log^2n$

---

unique(rk +1, rk + n + 1) 返回的是 rk 去重后的尾地址。
unique 并不真正把重复的元素删除，其实是，该函数把重复的元素一到后面去了，依然保存到了原数组中。
然后返回去重后最后一个元素的地址。
因为 unique 去除的是相邻的重复元素，所以一般用之前都会要排一下序。

线段树的大小与序列的最大值有关，所以我们要离散化，尽可能的缩小节点数量。
num 是去完重后的数的个数。线段树的大小即为 1 ~ num。

lower_bound(first, last, val) 返回一个非递减序列 [first, last) 中的第一个大于等于值 val的位置。

rk[] 存的是去完重的数的值
我们后面在空树基础上建树时，就用 lower_bound$(rk + 1, rk + num + 1, a[i]) - rk$ 找到在历史版本基础上新加入的数 a[i] 在 rk[] 中的位置，即后面建树时应 insert 的位置的编号（1 ~ num）。

---

扔板跑 …
求区间第 k 大值

#include <bits/stdc++.h>

using namespace std;
const int N = 2e5 + 5;

struct Node {
    int ls, rs, sum;
}T[N * 60]; // 数组一定要开够 

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

// 建空树 
int tot = 0;
void build(int& rt, int l, int r) {
    rt = ++ tot; T[rt].sum = 0;
    if (l == r) return ;
    int mid = (l + r) >> 1;
    build(T[rt].ls, l, mid), build(T[rt].rs, mid + 1, r);
}

// 用所有数的前缀建树 
void insert(int& rt, int pre, int l, int r, int rank) {
    rt = ++ tot;
    T[rt].ls = T[pre].ls; T[rt].rs = T[pre].rs;
    T[rt].sum = T[pre].sum + 1;
    if (l == r) return ;
    int mid = (l + r) >> 1;
    if (rank <= mid) insert(T[rt].ls, T[pre].ls, l, mid, rank);
    else insert(T[rt].rs, T[pre].rs, mid + 1, r, rank);
}

int query(int i, int j, int k, int l, int r) {
    if (l == r) return l;
    int t = T[T[i].ls].sum - T[T[j].ls].sum;
    int mid = (l + r) >> 1;
    if (k <= t) return query(T[i].ls, T[j].ls, k, l, mid);
    return query(T[i].rs, T[j].rs, k - t, mid + 1, r);
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++) scanf("%d", &a[i]), rk[i] = a[i];
    sort(rk + 1, rk + n + 1);
    int num = unique(rk + 1, rk + n + 1) - rk - 1; // num 是去完重后的数的个数
    // 此时 rk[] 存的是去完重的数的值

    tot = 0;
    build(root[0], 1, num);

    // 在空树基础上建树
    for (int i = 1; i <= n; i ++) // 注意这里是到 n，我们要把数列中的每一个数都要插入进去 
        insert(root[i], root[i - 1], 1, num, lower_bound(rk + 1, rk + num + 1, a[i]) - rk);

    while (m --) {
        int ql, qr, k;
        scanf("%d%d%d", &ql, &qr, &k);
        int ans = rk[query(root[qr], root[ql - 1], k, 1, num)];
        printf("%d\n", ans);
    }
    return 0;
}