2104 - K-th Number

原来是2104，刚才被坑死了。。

一道主席树区间k大的板题。

讲主席树的blog多了去了，随便去度娘翻翻就好，我就不多说了，直接挂代码。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
#define rep(i,j,k) for(i=j;i<=k;++i)
#define per(i,j,k) for(i=j;i>=k;--i)
#define ll long long
#define pli pair<ll,int>
#define mkp make_pair
#define X first
#define Y second
const int N=100005,inf=1000000000;
int n,m;
struct TREE{int lc,rc,sum;}T[5000000];
int tot,rt[N];
void ins(int &x,int y,int l,int r,int num){
	T[x=++tot]=T[num];
	++T[x].sum;
	if(l==r)return;
	int mid=l+r>>1;
	if(y>mid)ins(T[x].rc,y,mid+1,r,T[num].rc);
	else ins(T[x].lc,y,l,mid,T[num].lc);
}
int query(int x,int y,int k,int l,int r){
	if(l==r)return l;
	int mid=l+r>>1,tmp=T[T[y].lc].sum-T[T[x].lc].sum;
	return tmp<k?query(T[x].rc,T[y].rc,k-tmp,mid+1,r):query(T[x].lc,T[y].lc,k,l,mid);
}
int main(){
	int i,j,k,x;
	scanf("%d%d",&n,&m);
	rep(i,1,n){
		scanf("%d",&x);
		ins(rt[i],x,-inf,inf,rt[i-1]);
	}
	while(m--){
		scanf("%d%d%d",&i,&j,&k);
		printf("%d\n",query(rt[i-1],rt[j],k,-inf,inf));
	}
	return 0;
}

内存好像不太会算，但poj算内存的方式比较鬼畜，数组开太大是没有事的。这里就随便开个5000000吧。