90 lines
3.1 KiB
C++
90 lines
3.1 KiB
C++
#include <cstdio>
|
||
#include <algorithm>
|
||
using namespace std;
|
||
int read() {
|
||
int x = 0, f = 1;
|
||
char ch = getchar();
|
||
while (ch < '0' || ch > '9') {
|
||
if (ch == '-') f = -1;
|
||
ch = getchar();
|
||
}
|
||
while (ch >= '0' && ch <= '9') {
|
||
x = (x << 3) + (x << 1) + (ch ^ 48);
|
||
ch = getchar();
|
||
}
|
||
return x * f;
|
||
}
|
||
|
||
const int N = 2e5 + 10;
|
||
int n, m;
|
||
int a[N], b[N], bl; // b和bl是一组，用于离散化的数组,bl为b的数组中有用数字的个数，一般下标0不放东西
|
||
|
||
struct Node {
|
||
int l, r, cnt;
|
||
} tr[N << 5];
|
||
int root[N], idx;
|
||
|
||
//用于离散化的二分查找
|
||
int find(int x) {
|
||
return lower_bound(b + 1, b + 1 + bl, x) - b;
|
||
}
|
||
|
||
//经典的主席树插入
|
||
void insert(int &u, int l, int r, int x) {
|
||
tr[++idx] = tr[u]; //新开一个节点idx++,将新节点指向旧的tr[u]
|
||
tr[idx].cnt++; //新节点的cnt，因为多插入了一个数字，所以个数+1,这样处理的话，省去了pushup
|
||
u = idx; //因为是地址引用，需要回写u等于idx
|
||
|
||
if (l == r) return; //如果已经到了叶子节点，上面的操作就足够了，可以直接返回，否则需要继续向下递归
|
||
|
||
int mid = (l + r) >> 1;
|
||
if (x <= mid)
|
||
insert(tr[u].l, l, mid, x); //因为tr[u]进入本函数时，最先把旧的复制过来，所以tr[u].l也是上一个版本的左儿子节点
|
||
else
|
||
insert(tr[u].r, mid + 1, r, x);
|
||
}
|
||
// p:前面的版本，q:后面的版本,[l,r]：控制的范围
|
||
// k:要查找第k小的数字
|
||
int query(int p, int q, int l, int r, int k) {
|
||
if (l == r) return l;
|
||
int mid = (l + r) >> 1;
|
||
int cnt = tr[tr[q].l].cnt - tr[tr[p].l].cnt;
|
||
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() {
|
||
//文件输入输出
|
||
#ifndef ONLINE_JUDGE
|
||
freopen("P3834.in", "r", stdin);
|
||
#endif
|
||
n = read(), m = read();
|
||
for (int i = 1; i <= n; i++)
|
||
a[i] = b[i] = read();
|
||
|
||
//数据范围太大，直接建线段树会MLE，但是比较稀疏，可以离散化后用相对应的序号，数据量就小了
|
||
sort(b + 1, b + 1 + n);
|
||
bl = unique(b + 1, b + 1 + n) - b - 1; //离散化后共m个数字
|
||
|
||
// 0号版本没有内容时，主席树是不需要进行build的,强行build时，可能会有部分测试点TLE
|
||
// 0号版本有内容时，主席树是需要build的，不build，初始值无法给上
|
||
|
||
// 主席树的数字增加，每增加一个，就相当于增加了一个版本root[i]记录了版本i的根节点
|
||
for (int i = 1; i <= n; i++) {
|
||
root[i] = root[i - 1]; //开新版本号i,抄袭上一个版本i-1的根节点
|
||
insert(root[i], 1, bl, find(a[i])); //向版本i中增加find(a[i])的值
|
||
}
|
||
|
||
while (m--) {
|
||
int l, r, k;
|
||
l = read(), r = read(), k = read();
|
||
//采用类似于前缀的方法，对位相减，由于是动态开点，需要指明控制范围[1,bl]
|
||
//要查询的数字是k
|
||
printf("%d\n", b[query(root[l - 1], root[r], 1, bl, k)]);
|
||
}
|
||
|
||
return 0;
|
||
}
|