第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≤10⁵,M≤10⁴,|A[i]|≤10⁹

输入样例
7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3
输出样例
5
6
3

题目分析

主席树模板。
主席树本质上就是一个数组套线段树。

代码如下

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#define LL long long
#define ULL unsigned long long
#define PII pair<int,int>
#define x first
#define y second
using namespace std;
const int N=2e5+5,INF=0x3f3f3f3f;
struct Node{			//记录主席树节点
	int l,r;
	int size;
}tr[N*21];
int root[N],idx;		//root[]记录各个版本
int a[N];
vector<int> num;
int insert(int p,int l,int r,int x)			//插入数据
{
	int u=++idx;
	if(l==r)
	{
		tr[u].size=1;
		return u;
	}
	tr[u]=tr[p];
	int mid=l+r>>1;
	if(x<=mid) tr[u].l=insert(tr[p].l,l,mid,x);
	else tr[u].r=insert(tr[p].r,mid+1,r,x);
	tr[u].size=tr[tr[u].l].size+tr[tr[u].r].size;
	return u;
}
int query(int p,int q,int l,int r,int k)		//查询[l,r]内第k大数
{
	if(l==r) return l;
	int mid=l+r>>1;
	int size=tr[tr[p].l].size-tr[tr[q].l].size;
	if(size>=k) return query(tr[p].l,tr[q].l,l,mid,k);
	else return query(tr[p].r,tr[q].r,mid+1,r,k-size);
}
int main()
{
    int n,m;
    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());
	for(int i=1;i<=n;i++)
		a[i]=lower_bound(num.begin(),num.end(),a[i])-num.begin();
	
	for(int i=1;i<=n;i++)						//建树
		root[i]=insert(root[i-1],0,num.size()-1,a[i]);
	
	while(m--)
	{
		int l,r,k;
		scanf("%d %d %d",&l,&r,&k);
		printf("%d\n",num[query(root[r],root[l-1],0,num.size()-1,k)]);
	}
    return 0;
}