[主席树] 静态区间第k小 洛谷P3834

题目背景

这是个非常经典的主席树入门题——静态区间第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

说明

数据范围：

对于20%的数据满足：1 \leq N, M \leq 101≤N,M≤10

对于50%的数据满足：1 \leq N, M \leq 10^31≤N,M≤103

对于80%的数据满足：1 \leq N, M \leq 10^51≤N,M≤105

对于100%的数据满足：1 \leq N, M \leq 2\cdot 10^51≤N,M≤2⋅105

对于数列中的所有数a_iai​，均满足-{10}^9 \leq a_i \leq {10}^9−109≤ai​≤109

样例数据说明：

N=5，数列长度为5，数列从第一项开始依次为[25957, 6405, 15770, 26287, 26465 ][25957,6405,15770,26287,26465]

第一次查询为[2, 2][2,2]区间内的第一小值，即为6405

第二次查询为[3, 4][3,4]区间内的第一小值，即为15770

第三次查询为[4, 5][4,5]区间内的第一小值，即为26287

第四次查询为[1, 2][1,2]区间内的第二小值，即为25957

第五次查询为[4, 4][4,4]区间内的第一小值，即为26287

#include <bits/stdc++.h>
using namespace std;

const int mn = 2e5 + 10;

int a[mn], b[mn];
map<int, int> mp;

int rt[mn];
struct node
{
	int sz;
	int ls, rs;
} trie[40 * mn];    // 空间 2 * n * log(n)

int cnt;
void build(int l, int r, int &now)
{
	now = ++cnt;    // 新建节点 编号 = ++cnt
	
	if (l == r)
		return;
		
	int mid = (l + r) / 2;
	build(l, mid, trie[now].ls);
	build(mid + 1, r, trie[now].rs);
}

void inser(int l, int r, int last, int &now, int ID)
{
    now = ++cnt;    // 新建节点
    trie[now] = trie[last];     // 复制上一棵树对应节点信息
    trie[now].sz++;     // 更改信息
    
    if (l == r)
        return;
        
    int mid = (l + r) / 2;
    if (ID <= mid)
        inser(l, mid, trie[last].ls, trie[now].ls, ID);
    else 
        inser(mid + 1, r, trie[last].rs, trie[now].rs, ID);
}

/// 区间第k大
int query(int a, int b, int l, int r, int k)
{
    if (l == r)
        return l;
    
    int mid = (l + r) / 2;
    int temp = trie[trie[b].ls].sz - trie[trie[a].ls].sz;   // 两树对应节点包含的数的数量之差
    if (k <= temp)
        return query(trie[a].ls, trie[b].ls, l, mid, k);
    else 
        return query(trie[a].rs, trie[b].rs, mid + 1, r, k - temp);
}

/*
/// 区间内 <= k的数量
int query(int a, int b, int l, int r, int k)
{
	if (l == r) return tr[b].sz - tr[a].sz;
	int mid = (l + r) >> 1;
	int temp = tr[tr[b].ls].sz - tr[tr[a].ls].sz;
	if (k <= mid) return query(tr[a].ls, tr[b].ls, l, mid, k);
	else return temp + query(tr[a].rs, tr[b].rs, mid + 1, r, k);
}
*/

int main()
{	
	int n, m;
	scanf("%d %d", &n, &m);
	build(1, n, rt[0]);     /// 建立初始空树
	
	for (int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
        b[i] = a[i];
    }
	sort(b + 1, b + n + 1);
	for (int i = 1; i <= n; i++)
        mp[b[i]] = i;
    /// 离散化
	
	for (int i = 1; i <= n; i++)
		inser(1, n, rt[i - 1], rt[i], mp[a[i]]);
    /// 在第 mp[a[i]] (第k小) 的位置 插入第 i 个数
	
	while (m--)
	{
		int L, R, k;
		scanf("%d %d %d", &L, &R, &k);
		
		int ans = query(rt[L - 1], rt[R], 1, n, k);
		/// rt[L- 1] = 插入前 L - 1 个数后的树, rt[R] = 插入第 R 个数后的树
		/// 两棵树比较 取两棵树相差的数中的第 k 小的数
		cout << b[ans] << endl;
	}
	
	return 0;
}