主席树（静态）入门篇

主席树入门：

静态（只有查询，没有修改）：

引入：为什么要引入主席树呢，我们先从一道题进行引入。这样的：就是给了一个序列，我要进行若干次的查询，就是给定区间[L,R]，查询这个区间中第k大的数字。显然，多次查询就不大好办了，显然，我们第一想到的就是能否利用前缀和的思想。

那么我们就开门见山，原序列是a【】，假设a序列离散化完了，然后让rt【o】这棵树，代表当前插入了a[1]-a[o]这些数字后的这样的一颗线段树。我们建立n棵线段树。

那么想一下，如果让sum【节点号】代表这个节点代表区间的插入数字的个数，（就是类似于线段树中维护的那个值），假设我要求【L,R】区间里的第k小的，那么我们此时就想啊，我们建立的第R棵线段树，和第L棵线段树，对应位置上的数字一减是不是就是我在【L，R】中的所有数字哇，然后就是正常的查询第k个小的就好了。

但是问题来了，正常建立这么多线段树内存肯定是要炸的。于是有人就画了画就发现了，咦，我的第 i 颗线段树比第 i-1 颗线段树不就只增加了一个点而已嘛，就是一条链上的节点，那么影响的节点就是logn个。不变的节点就不让他变了，节点重复利用嘛！于是就出现了好多博客里的那种复杂的树。（其实并不复杂，但博主当时看了好一会儿）。

比如说（以区间第k大为例hdu2665题目戳这里http://acm.hdu.edu.cn/showproblem.php?pid=2665）:

设n = 4，q= 1;

4个数分别为4， 1， 3 ，2;

ql = 1, qr = 3, k = 2;

 

emmm，盗完图了。

图的来源：

https://www.cnblogs.com/zyf0163/p/4749042.html

看不懂就数节点的编号。会发现这个节点改变的话就是会多加一个节点的。我们怎么写呢？感觉不会写欸：

void update(int& o, int l, int r, int last, int p){
    o = ++ tot;
    ls[o] = ls[last];
    rs[o] = rs[last];
    sum[o] = sum[last] + 1;
    if(l == r) return;
    int m = (l + r) >> 1;
    if(p <= m)  update(ls[o], l, m, ls[last], p);
    else update(rs[o], m + 1, r, rs[last], p);
}

看这个代码，就是用last代表第i-1棵树，o是当前的第i棵树，然后我们走的路径上的点肯定是新加的。而原来，共享的就不变就好了。

注意：&o的意思是取地址，从哪来的就改变哪个值。这样rt【o】实际上每一次都是在改变的。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 100000 + 5;

int a[N], b[N], rt[N * 20], ls[N * 20], rs[N * 20], sum[N * 20];

int n, k, tot, sz, ql, qr, x, q, T;

void Build(int& o, int l, int r){
    o = ++ tot;
    sum[o] = 0;
    if(l == r) return;
    int m = (l + r) >> 1;
    Build(ls[o], l, m);
    Build(rs[o], m + 1, r);
}
void update(int& o, int l, int r, int last, int p){
    o = ++ tot;
    ls[o] = ls[last];
    rs[o] = rs[last];
    sum[o] = sum[last] + 1;
    if(l == r) return;
    int m = (l + r) >> 1;
    if(p <= m)  update(ls[o], l, m, ls[last], p);
    else update(rs[o], m + 1, r, rs[last], p);
}

int query(int ss, int tt, int l, int r, int k){
    if(l == r) return l;
    int m = (l + r) >> 1;
    int cnt = sum[ls[tt]] - sum[ls[ss]];
    if(k <= cnt) return query(ls[ss], ls[tt], l, m, k);
    else return query(rs[ss], rs[tt], m + 1, r, k - cnt);
}

void work(){
    scanf("%d%d%d", &ql, &qr, &x);
    int ans = query(rt[ql - 1], rt[qr], 1, sz, x);
    printf("%d\n", b[ans]);
}

int main(){
    scanf("%d", &T);
    while(T--){
        scanf("%d%d", &n, &q);
        for(int i = 1; i <= n; i ++) scanf("%d", a + i), b[i] = a[i];
        sort(b + 1, b + n + 1);
        sz = unique(b + 1, b + n + 1) - (b + 1);
        tot = 0;
        Build(rt[0],1, sz);
        //for(int i = 0; i <= 4 * n; i ++)printf("%d,rt =  %d,ls =  %d, rs = %d, sum = %d\n", i, rt[i], ls[i], rs[i], sum[i]);
        for(int i = 1; i <= n; i ++)a[i] = lower_bound(b + 1, b + sz + 1, a[i]) - b;
        for(int i = 1; i <= n; i ++)
        {
          ///  cout<<rt[i]<<"   &&&   "<<rt[i-1]<<endl;
            update(rt[i], 1, sz, rt[i - 1], a[i]);
        }
        ///for(int i = 0; i <= 5 * n; i ++)printf("%d,rt =  %d,ls =  %d, rs = %d, sum = %d\n", i, rt[i], ls[i], rs[i], sum[i]);
        while(q --)work();
    }
    return 0;
}