poj2104 K-th NumberK-th Number【主席树】

链接：http://poj.org/problem?id=2104

题意：给你n个小于10^9的数，m个询问。问你l到r中第k小的数是什么。

分析：主席树入门题

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define Mn 200005
#define Mm 2000005
#define mod 1000000007
#define CLR(a,b) memset((a),(b),sizeof((a)))
#define CPY(a,b) memcpy ((a), (b), sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)|1
using namespace std;
typedef long long ll;
int read() {
    char c;
    int ans=0,f=1;c=getchar();
    while(c<'0'||c>'9') {if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9') {ans=ans*10+c-'0';c=getchar();}
    return  ans*f;
}
int tot=0;
int root[Mn],ls[Mm],rs[Mm],sum[Mm];
void Insert(int l,int r,int x,int &y,int val) {
    y=++tot;
    sum[y]=sum[x]+1;
    if(l==r) return ;
    ls[y]=ls[x],rs[y]=rs[x];
    int mid=(l+r)>>1;
    if(val<=mid) Insert(l,mid,ls[x],ls[y],val);
    else Insert(mid+1,r,rs[x],rs[y],val);
}
int ask(int l,int r,int x,int y,int k) {
    if(l==r) return l;
    int mid=(l+r)>>1;
    if(sum[ls[y]]-sum[ls[x]]>=k) return ask(l,mid,ls[x],ls[y],k);
    else return ask(mid+1,r,rs[x],rs[y],k-(sum[ls[y]]-sum[ls[x]]));
}
int a[Mn],b[Mn],cnt;
int find(int x){
    int l=1,r=cnt;
    while (l<=r){
        int mid=(l+r)>>1;
        if (b[mid]<x) l=mid+1;
        else r=mid-1;
    }
    return l;
}
int main() {
    tot=0;
    int n=read(),m=read();
    for(int i=1;i<=n;i++) a[i]=read(),b[i]=a[i];
    sort(b+1,b+1+n);
    cnt=unique(b+1,b+1+n)-b;
    //for(int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+cnt+1,a[i])-b;
    for(int i=1;i<=n;i++) Insert(1,cnt,root[i-1],root[i],find(a[i]));
    for(int i=1;i<=m;i++) {
        int l=read(),r=read(),k=read();
        printf("%d\n",b[ask(1,cnt,root[l-1],root[r],k)]);
    }
    return 0;
}