主席树模板

今天学习了主席树，我觉得我可以去黑入美国国防部的网络了。
所谓主席树，它的真正含义和主席两个字完全没有联系。
主席树的作用是保存历史版本的线段树。

不知为何我对这张图片情有独钟。
划圈的部分是单点修改后的新的线段树。

下面代码是一道主席树模板题，求区间第k小的数的值。
此题思路就是每次插入一个数，就新建一颗权值线段树，变成主席树。
输入x到y求第k小的数，就是用第y个版本的线段树减去第x-1个版本的线段树，这样所剩下的权值线段树就是x到y这一区间上的数组成的权值线段树了，然后查询第k小。
就很妙。

//#pragma GCC optimize(3,"Ofast","inline")
//#include<unordered_map>
#include<cstdio>
#include<iostream>
#include<cmath>
#include<functional>
#include<cstring>
#include<string>
#include<cstdlib>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#include<stack>
#include<vector>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f;
int mod=1000000000;
const int maxn=2e5+7;
int n,m,cnt,root[maxn],a[maxn];
//n是几个点，m是几次查询，cnt就是个计数的，类似于是邻接表里的cnt。
//root数组储存每个版本线段树的根节点编号，类似于邻接表head数组，a数组储存输入的n个数。
struct node
{
    int l,r,sum;
}t[maxn*40];//线段树的结构体
vector<int >v;//此物是stl的容器，在这里用于离散化。
int getid(int x)
{
    return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}//这个函数是找出以排好序的容器中x在哪个位置，update的时候用
void update(int l,int r,int &now,int last,int id)
{
    t[++cnt]=t[last];//新建一个点等于上一个版本的点
    t[cnt].sum++;//修改sum
    now=cnt;//带&修改上次递归的点的左或者右孩子
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(id<=mid)update(l,mid,t[now].l,t[last].l,id);//套路安排左边
    else update(mid+1,r,t[now].r,t[last].r,id);//套路安排右边
}
int query(int l,int r,int x,int y,int k)//查询
{
    if(l==r)return l;
    int mid=(l+r)>>1;
    int sum=t[t[y].l].sum-t[t[x].l].sum;//这里找出左孩子所代表的区间里有多少个元素
    if(sum>=k) return query(l,mid,t[x].l,t[y].l,k);//如果k小于这个数，那么要找的数就在左边
    else return query(mid+1,r,t[x].r,t[y].r,k-sum);//反之，找右边，那么在右边的话就是查询第k-sum小的数了
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        v.push_back(a[i]);
    }//输入
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());//离散化
    for(int i=1;i<=n;i++)
        update(1,n,root[i],root[i-1],getid(a[i]));//插入
    int l,r,q;
    while(m--)//每次查找
    {
        scanf("%d%d%d",&l,&r,&q);
        printf("%d\n",v[query(1,n,root[l-1],root[r],q)-1]);
    }
    return 0;
}