P1533 可怜的狗狗 [主席树模板]

最新推荐文章于 2020-03-14 10:36:32 发布

原创最新推荐文章于 2020-03-14 10:36:32 发布 · 164 阅读

0 ·

CC 4.0 BY-SA版权

主席树专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种使用线段树进行离线查询的方法，通过预处理建立数据结构，实现快速查询区间内的第k小元素。具体实现包括读取输入、排序、离散化、构建线段树、插入元素和查询操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门

#include<bits/stdc++.h>
#define N 300050
using namespace std;
int a[N],b[N],n,m,rt[N],tot;
struct Node{int l,r,siz;}t[N*40];
int read(){
	int cnt=0; char ch=0;
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
	return cnt;
}
void build(int &x,int l,int r){
	x = ++tot;
	if(l==r) return;
	int mid = (l+r)>>1;
	build(t[x].l,l,mid);
	build(t[x].r,mid+1,r);
}
void Insert(int &x,int l,int r,int last,int pos){
	x = ++tot;
	t[x].l = t[last].l; t[x].r = t[last].r;
	t[x].siz = t[last].siz + 1;
	if(l==r) return;
	int mid = (l+r)>>1;
	if(pos<=mid) Insert(t[x].l,l,mid,t[last].l,pos);
	else Insert(t[x].r,mid+1,r,t[last].r,pos);
}
int quary(int L,int R,int l,int r,int k){
	if(l==r) return l;
	int mid = (l+r)>>1;
	int cnt = t[t[R].l].siz - t[t[L].l].siz;
	if(k<=cnt) return quary(t[L].l,t[R].l,l,mid,k);
	else return quary(t[L].r,t[R].r,mid+1,r,k-cnt); 
}
int main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++) a[i] = b[i] = read();
	sort(b+1,b+n+1); int siz = unique(b+1,b+n+1) - (b+1);
	for(int i=1;i<=n;i++) a[i] = lower_bound(b+1,b+siz+1,a[i]) - b;
	build(rt[0],1,siz);
	for(int i=1;i<=n;i++) Insert(rt[i],1,siz,rt[i-1],a[i]);
	for(int i=1;i<=m;i++){
		int x=read(),y=read(),k=read();
		printf("%d\n",b[quary(rt[x-1],rt[y],1,siz,k)]);
	} return 0;
}