p3834——主席树模板题

最近遇到好多题目都是关于主席树的，但是菜菜没有学过，因此学习了一下主席树的模板。

这个up大大讲的特别好！参考：网页链接
https://www.bilibili.com/video/BV1C4411u7rK?from=search&seid=9313032007306021838

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+10;
int n,m;
int cnt;
struct node
{
    int L,R;
    int sum;
    node()
    {
        sum=0;
    }
} Tree[maxn*40];
vector<int>v;
int getid(int x)
{
    //尝试使用离散化标准高级写法 二分！
    return lower_bound(v.begin(),v.end(), x)-v.begin()+1;
}
int root[maxn];//每棵线段树的根节点
int a[maxn];//原数组
//感觉init()可有可无的说
//void init()
//{
//    cnt=0;
//    root[0]=0;
//    Tree[0].L=Tree[0].R=Tree[0].sum=0;
//}

void update(int num, int pre,int &now, int l, int r)
{
    //将Tree[pre]节点是传入的旧的节点，原封不动的先赋值给Tree[cnt];
    //再修改Tree[cnt],即需要修改的新的节点
    //使用引用做到父节点的结构体中也更新;
    //因此 解释图中新旧节点的联系 实际上就是动态开辟一个新的节点
    Tree[++cnt]=Tree[pre];
    now=cnt;
    Tree[now].sum++;
    if(l==r)return ;
    int mid=(l+r)>>1;
    if(num<=mid)update(num,Tree[pre].L,Tree[now].L,l, mid);
    else update(num,Tree[pre].R,Tree[now].R,mid+1,r);
}
int query(int i, int j, int k, int l, int r)
{
    //第j棵树-第i棵树左子树的值=在i,j区间内权值的个数
    //i和j是两个版本的权值线段树 是同步的 只是插入l-r区间前和插入后的差别
    int d=Tree[Tree[j].L].sum-Tree[Tree[i].L].sum;
    if(l==r)return l;//如果左=右，则就是这个数
    int mid=(l+r)>>1;
    if(k<=d)return query (Tree[i].L, Tree[j].L,k,l,mid);
    else return query(Tree[i].R,Tree[j].R, k-d,mid+1,r);
}
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(getid(a[i]), root[i-1],root[i],1,n);
    }
    int lft,rgt,k;

    for(int i=1;i<=m;i++){
        scanf("%d %d %d",&lft,&rgt, &k);
        //注意下标-1
        printf("%d\n",v[query(root[lft-1],root[rgt],k,1,n)-1]);
    }


    return 0;
}