AcWing 255. 第K小数

给定长度为 N 的整数序列 A，下标为 1∼N。

现在要执行 M 次操作，其中第 i 次操作为给出三个整数 li,ri,ki，求 A[li],A[li+1],…,A[ri] (即 A 的下标区间 [li,ri])中第 ki 小的数是多少。

输入格式
第一行包含两个整数 N 和 M。

第二行包含 N 个整数，表示整数序列 A。

接下来 M 行，每行包含三个整数 li,ri,ki，用以描述第 i 次操作。

输出格式
对于每次操作输出一个结果，表示在该次操作中，第 k 小的数的数值。

每个结果占一行。

数据范围
N≤105,M≤104,|A[i]|≤109
输入样例：
7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3
输出样例：
5
6
3

p 为 某节点的 地址

tr[p] . l 表示 左子树的数量
tr[p] . r表示 右子树的数量
注意： 左右子树的范围

这道题是可持续化线段树， 我的理解是线段树在不同节点的不同版本。
每次遍历一个点就会创建新的线段树，线段树的长度应为对原有数据离散化的长度(num.size())。每次会占用 4 * num.size() 大小的数组空间 ，一共会有 N 大小的 节点，在开数组的时候需要计算一下占用的空间大小 。

build操作，类似于 线段树的 build 的操作，左子树的含义是当前节点范围的左半部分 ， 右子树以此类推 。主要是为了建立一个参照对象，方便后续的insert(modify)函数。

Insert操作 ，类似于 线段的modify的操作 ，基本上是复制前一个根节点，操作基本上完全一样，但是在某个节点上的个数不同，需要在这个 l = r && r = x的点的个数加 1 ， 然后pushup操作更新节点， 最后返回根节点的地址。（类似于数据结构相同，点的个数不同）

query操作， 注意还是类似于二分 ， 如果小于值的一半的个数大于k，说明 我们要求的k在大于值的一半的部分， 然后二分的不是下标，而是值的范围。

代码如下：

#include<iostream>
#include<algorithm>
#include<vector>

using namespace std;

const int N = 1e5 + 10 ;

struct Node{
    int l , r ;
    int cnt ; 
} tr[4 * N + N * 17 ];

int n , m ;
int root[N];
int a[N];
vector<int> num ;
int idx ; 

int find(int x){
    return lower_bound( num.begin() , num.end() , x ) - num.begin() ;
}

int build(int l , int r){
   int q = ++idx ;
   if(l == r) return q;
   int mid = l + r >> 1 ;
   tr[q].l = build( l , mid );
   tr[q].r = build( mid + 1 , r );
   return q ;
}

int insert(int p , int l , int r , int x){//这里的 l ,r 不是数组的下标，而是值的范围大小
    int q = ++idx;
    if(l == r){
        tr[q].cnt++;
        return q;
    } 
    tr[q] = tr[p];
    int mid = l + r >> 1; 
    if(x <= mid ) tr[q].l = insert( tr[p].l , l , mid , x );
    else tr[q].r = insert( 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 p , int q , int l , int r , int  k)//这里的 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[p].l , tr[q].l , l , mid , k );
    else return query( tr[p].r , tr[q].r , mid + 1 , r , k - cnt );
}

int main(void){
    scanf("%d%d",&n,&m);
    for(int i = 1 ; i <= n ; i ++ )
    {
        scanf( "%d" , &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] = insert(root[i - 1] , 0 , num.size() - 1 , find(a[i]));
    }
    
    while( m -- ){
        int l , r , k ;
        scanf("%d%d%d", &l , &r , &k );
        printf("%d\n",num[query(root[l - 1], root[r] , 0 , num.size() - 1 , k )]);
    }
}