K-th Number POJ - 2104（主席树）

题目链接

题意：求静态区间第K大

验板子 

注释的地方是我默写过程中记不清、出错的或者我自己觉得要注意的

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int N=1e5+5;
int a[N];
vector<int>v;
inline int getid(int x){
	return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}
struct{
	int l,r,sum;
}hjt[N*40];
int cnt,root[N];
void insert(int l,int r,int pre,int& now,int val){
	hjt[++cnt]=hjt[pre];
	now=cnt;
	hjt[now].sum++;
	if(l==r) return;
	int m=l+r>>1;
	if(val<=m) insert(l,m,hjt[pre].l,hjt[now].l,val);//pre.l now.l
	else insert(m+1,r,hjt[pre].r,hjt[now].r,val);//pre.r now.r
}
int query(int l,int r,int L,int R,int k){
	if(l==r) return l;
	int tmp=hjt[hjt[R].l].sum-hjt[hjt[L].l].sum;
	int m=l+r>>1;
	if(k<=tmp) return query(l,m,hjt[L].l,hjt[R].l,k);
	else return query(m+1,r,hjt[L].r,hjt[R].r,k-tmp);//k-tmp
}
int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i){
		scanf("%d",&a[i]);
		v.push_back(a[i]);
	}
	sort(v.begin(),v.end());
	v.erase(unique(v.begin(),v.end()),v.end());
	for(int i=1;i<=n;++i){
		insert(1,n,root[i-1],root[i],getid(a[i]));//v[getid(a[i])]=a[i] //root[]
	}
	while(m--){
		int l,r,k;
		scanf("%d%d%d",&l,&r,&k);
		printf("%d\n",v[query(1,n,root[l-1],root[r],k)-1]);//root[]
	}
}