POJ 2104 K-th Number 题解&代码

你说什么？和HDU题目一样？不不不这才不是多组数据= =
其实重点在于我终于知道为什么我一直TLE了…map被卡了那个萌萌哒logn
所谓STL一时爽全家火葬场，改了数组记录之后过得轻轻松松
代码如下【一会去把hdu的排序也改一下重交试试

#include<iostream>
#include<algorithm>
#include<map>
#include<string.h>
#include<stdio.h>
using namespace std;
const int maxn=100005;
int T,m,n,L,R,K,tot,cnt,line[maxn],vis[maxn],p[maxn],r[maxn];
int tree[maxn*20],son[2][maxn*20],rt[maxn];
void add_tree(int rl,int rn,int l,int r,int value)
{
    tree[rn]=tree[rl]+1;
    if(l==r)return;
    int mid=(l+r)/2;
    if(value>mid)son[0][rn]=son[0][rl],son[1][rn]=++tot,add_tree(son[1][rl],son[1][rn],mid+1,r,value);
    else son[1][rn]=son[1][rl],son[0][rn]=++tot,add_tree(son[0][rl],son[0][rn],l,mid,value);
}
int query(int rl,int rn,int l,int r,int k)
{
    if(l==r)return vis[l];
    int mid=(l+r)/2;
    if(k>tree[son[0][rn]]-tree[son[0][rl]])return query(son[1][rl],son[1][rn],mid+1,r,k-tree[son[0][rn]]+tree[son[0][rl]]);
    else return query(son[0][rl],son[0][rn],l,mid,k);
}
bool cmp(int a,int b)
{
    return line[a]<line[b];
}
int main(void)
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&line[i]);
        p[i]=i;
    }
    sort(p+1,p+1+n,cmp);
    for(int i=1;i<=n;i++)
    {
        vis[i]=line[p[i]];
        r[p[i]]=i;
    }
    for(int i=1;i<=n;i++)
    {
        rt[i]=++tot;
        add_tree(rt[i-1],rt[i],1,n,r[i]);
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&L,&R,&K);
        printf("%d\n",query(rt[L-1],rt[R],1,n,K));
    }
    return 0;
}