luoguP3834 【模板】可持久化线段树 1（主席树）

题目背景

这是个非常经典的主席树入门题——静态区间第K小

数据已经过加强，请使用主席树。同时请注意常数优化

题目描述

如题，给定N个正整数构成的序列，将对于指定的闭区间查询其区间内的第K小值。

输入输出格式

输入格式：
第一行包含两个正整数N、M，分别表示序列的长度和查询的个数。
第二行包含N个正整数，表示这个序列各项的数字。
接下来M行每行包含三个整数 l, r, kl,r,k , 表示查询区间[l, r][l,r]内的第k小值。

输出格式：
输出包含k行，每行1个正整数，依次表示每一次查询的结果

输入输出样例

输入样例#1：
5 5
25957 6405 15770 26287 26465
2 2 1
3 4 1
4 5 1
1 2 2
4 4 1

输出样例#1：
6405
15770
26287
25957
26287

样例数据说明：

N=5，数列长度为5，数列从第一项开始依次为
25957, 6405, 15770, 26287, 26465
第一次查询为[2, 2]区间内的第一小值，即为6405
第二次查询为[3, 4]区间内的第一小值，即为15770
第三次查询为[4, 5]区间内的第一小值，即为26287
第四次查询为[1, 2]区间内的第二小值，即为25957
第五次查询为[4, 4]区间内的第一小值，即为26287

分析：
练板子

tip

离散化注意

sort(a+1,a+1+n,cmp);for (int i=1;i<=n;i++){    num[a[i].id]=i;    V[i]=a[i].x;}

在build的时候now和top的关系不要搞混

不要忘了元素个数的增加

top++;t[top]=t[now];now=top;t[now].sum++;    //元素个数

询问

int ask(int x,int y,int l,int r,int k){    if (l==r) return l;   //权值线段树，位置就是离散值     int tmp=t[t[y].l].sum-t[t[x].l].sum;    int mid=(l+r)>>1;     if (tmp>=k) return ask(t[x].l,t[y].l,l,mid,k);    else return ask(t[x].r,t[y].r,mid+1,r,k-tmp);}

这里写代码片#include<cstdio>#include<iostream>#include<cstring>#include<algorithm>using namespace std;const int N=200001;struct node{    int l,r,sum;};node t[N*40];int top=0,n,root[N],num[N],V[N],m;struct nd{    int x,id;};nd a[N];int cmp(const nd &a,const nd &b){    return a.x<b.x;}void build(int &now,int l,int r,int z){    top++;    t[top]=t[now];    now=top;    t[now].sum++;    //元素个数    if (l==r) return;    int mid=(l+r)>>1;    if (z<=mid) build(t[now].l,l,mid,z);    else build(t[now].r,mid+1,r,z); }int ask(int x,int y,int l,int r,int k){    if (l==r) return l;   //权值线段树，位置就是离散值     int tmp=t[t[y].l].sum-t[t[x].l].sum;    int mid=(l+r)>>1;     if (tmp>=k) return ask(t[x].l,t[y].l,l,mid,k);    else return ask(t[x].r,t[y].r,mid+1,r,k-tmp);}int main(){    scanf("%d%d",&n,&m);    for (int i=1;i<=n;i++)        scanf("%d",&a[i].x),a[i].id=i;    sort(a+1,a+1+n,cmp);    for (int i=1;i<=n;i++)    {        num[a[i].id]=i;        V[i]=a[i].x;    }    for (int i=1;i<=n;i++)    {        root[i]=root[i-1];        build(root[i],1,n,num[i]);    }    for (int i=1;i<=m;i++)    {        int x,y,k;        scanf("%d%d%d",&x,&y,&k);        printf("%d\n",V[ask(root[x-1],root[y],1,n,k)]);    }    return 0;}