POJ 2104 K-th Number

题目链接：传送门

题目大意：长为 n 的数组，m 次询问，每次问区间 [l,r] 内第 k 小的数。

题目思路：主席树

应该是按权值建立线段树，先将权值离散化，然后利用前缀和的思想来解决。

　　　　　主席树还是有点难理解，，，

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <cctype>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <climits>
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define fi first
#define se second
#define ping(x,y) ((x-y)*(x-y))
#define mst(x,y) memset(x,y,sizeof(x))
#define mcp(x,y) memcpy(x,y,sizeof(y))
using namespace std;
#define gamma 0.5772156649015328606065120
#define MOD 1000000007
#define inf 0x3f3f3f3f
#define N 100005
#define maxn 30010
typedef pair<int,int> PII;
typedef long long LL;
LL read(){
LL 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-'0';ch=getchar();}
return x*f;
}

int n,m,k,sz,L,R;
int a
,b
,c
;
int root
;
struct Seg{
int l,r,v;
}seg[N*30];
void update(int &rot,int rt,int l,int r){
seg[++sz]=seg[rot],rot=sz;
++seg[rot].v;
if(l==r)return;
int mid=l+r>>1;
if(L<=mid)update(seg[rot].l,lson);
else update(seg[rot].r,rson);
}
int query(int rt,int l,int r,int k){
if(l==r)return a[l];
int num=seg[seg[R].l].v-seg[seg[L].l].v;
int mid=l+r>>1;
if(num>=k){
L=seg[L].l;R=seg[R].l;
return query(lson,k);
}
else{
L=seg[L].r;R=seg[R].r;
return query(rson,k-num);
}
}
int main(){
int i,j,group,x,y;
n=read();m=read();
for(i=1;i<=n;++i)a[i]=read(),c[i]=b[i]=a[i];
sort(a+1,a+n+1);
int _n=unique(a+1,a+n+1)-a;
for(i=1;i<=n;++i)b[i]=lower_bound(a+1,a+_n+1,b[i])-a;
for(i=1;i<=n;++i)L=b[i],update(root[i]=root[i-1],1,1,n);
while(m--){
x=read(),y=read(),k=read();
L=root[x-1];R=root[y];
printf("%d\n",query(1,1,n,k));
}
return 0;
}