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

来源：互联网发布：卡廷森林知乎编辑：程序博客网时间：2024/06/02 14:56

链接：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)|1using 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;}

0 0