poj2828（线段树查找序列第k小的值）

题目链接：https://vjudge.net/problem/POJ-2828

题意：有n个人，依次给出这n个人进入队列时前面有多少人p[i]，和它的权值v[i]，求最终队列的权值序列。

思路：基本类似于poj2182，简化题意后即为求序列1..n中第k小的值的问题。读入数据量比较大，最好读入优化。我们从n..1逆序遍历，则可以确认最后一个人的最终位置为p[n]+1，然后从序列中删除p[n]+1，继续操作倒数第二个...这一操作可以通过线段树来完成。线段树的结点包括3个值：l（区间左端点），r（区间右端点），len（区间剩余编号数）。每次询问维护len值。时间复杂度为O(Tnlogn)。

AC代码：

#include<cstdio>
#include<cctype>
using namespace std;
const int maxn=;

inline int read(){
    int x=,f=;char c=;
    while(!isdigit(c)) {f|=c=='-';c=getchar();}
    while(isdigit(c)) x=(x<<)+(x<<)+(c^),c=getchar();
    return f?-x:x;
}

struct node{
    int l,r,len;
}tr[*maxn];

int p[maxn],v[maxn],n,ans[maxn];

void build(int v,int l,int r){
    tr[v].l=l,tr[v].r=r,tr[v].len=r-l+;
    if(l==r) return;
    int mid=(l+r)>>;
    build(*v,l,mid);
    build(*v+,mid+,r);
}

int query(int v,int k){
    --tr[v].len;
    if(tr[v].l==tr[v].r) return tr[v].l;
    if(k<=tr[*v].len) return query(*v,k);
    else return query(*v+,k-tr[*v].len);
}

int main(){
    while(~scanf("%d",&n)){
        build(,,n);
        for(int i=;i<=n;++i)
            p[i]=read(),v[i]=read();
        for(int i=n;i>=;--i)
            ans[query(,p[i]+)]=v[i];
        for(int i=;i<=n;++i)
            printf("%d ",ans[i]);
        printf("\n");
    }
    return ;
}