树-主席树

主席树

主席树又被称为可持久化权值线段树，是可以记录历史版本的一棵权值线段树。
对于每次新插入的点，我们新建一个根节点，然后复制上一个根节点建立的树，在此基础上把这个节点添加上去，就完成了当前版本的权值线段树。
主席树最经典的用法就是求区间内第k大数，我们知道权值线段树可以用来求解整个序列第k大数，但是如果换成区间的话就没有办法了，但是在主席树上，由于每个点新建的时候都是有自己的版本，那么root[r]则表示从1建立到r这个点的树，root[l-1]表示从1建立到l-1这个点的数，那么它们相减便是区间[l,r]内的数，再利用权值线段树的特性求解即可。

例题1 ：第k大数 (POJ2104)

您在 Macrohard 公司的数据结构部门工作。在你之前关于键插入的任务失败后，你被要求编写一个新的数据结构，它能够在数组段中快速返回 k 阶统计信息。
也就是说，给定一个由不同整数组成的数组 a[1…n]，您的程序必须以下列形式回答一系列问题 Q(i, j, k)：“a 中的第 k 个数是多少？ [i…j] 段，如果此段已排序？”
例如，考虑数组 a = (1, 5, 2, 6, 3, 7, 4)。设问题为 Q(2, 5, 3)。段 a[2…5] 是 (5, 2, 6, 3)。如果我们对这个段进行排序，我们得到 (2, 3, 5, 6)，第三个数字是 5，因此问题的答案是 5。

输入格式

第一行包含两个整数 n 和 q。
第二行包含 n 个整数，表示整数数列。
接下来 q 行，每行包含三个整数 l，r，k，表示一个询问的区间[l,r]内第k大的数。

输出格式

共 q 行，每行输出一个询问的结果。

主席树模板

//主席树-区间第k大数
#include "bits/stdc++.h"
using namespace std;
const int N=100010,M=10010;
int n,m,a[N],root[N],idx;
vector<int> num;
struct node
{
    int l,r,cnt;
}tr[20*N];
int find(int x)//找到离散化对应原数组的值
{
    return lower_bound(num.begin(),num.end(),x)-num.begin();
}
int build(int l,int r)
{
    int p=++idx;
    if(l==r) return p;
    int mid=l+r>>1;
    tr[p].l=build(l,mid),tr[p].r=build(mid+1,r);
    return p;
}
int update(int p,int l,int r,int x)
{
    int q = ++ idx;
    tr[q] = tr[p];//新树先复制旧树
    if (l == r)
    {
        tr[q].cnt ++ ;
        return q;
    }
    //再给新树添加单个节点
    int mid = l + r >> 1;
    if (x <= mid) tr[q].l = update(tr[p].l, l, mid, x);
    else tr[q].r = update(tr[p].r, mid + 1, r, x);
    tr[q].cnt = tr[tr[q].l].cnt + tr[tr[q].r].cnt;
    return q;
}
int query(int q, int p, int l, int r, int k)//用root[r]版本减去root[l-1]版本便可以知道在[l,r]之间的节点个数
{
    if (l == r) return r;
    int cnt = tr[tr[q].l].cnt - tr[tr[p].l].cnt;
    int mid = l + r >> 1;
    if (k <= cnt) return query(tr[q].l, tr[p].l, l, mid, k);
    else return query(tr[q].r, tr[p].r, mid + 1, r, k - cnt);
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) 
    {
        cin>>a[i];
        num.push_back(a[i]);//离散化
    }
    sort(num.begin(),num.end());
    num.erase(unique(num.begin(),num.end()),num.end());//去掉重复元素
    root[0]=build(0,num.size()-1);//建立无节点的数
    for(int i=1;i<=n;i++) root[i]=update(root[i-1],0,num.size()-1,find(a[i]));//按照版本依次记录更新根节点
    while(m--)
    {
        int l,r,k;
        cin>>l>>r>>k;
        cout<<num[query(root[r],root[l-1],0,num.size()-1,k)]<<endl;
    }
}