首页 > 代码库 > poj 2828 Buy Tickets

poj 2828 Buy Tickets

http://poj.org/problem?id=2828

 

题意:n个人排队,每个人都有一个pos(位置)和一个val值,求队伍的最终状态。

思路: 线段树 单点更新。以叶子节点代表每个人的位置,

从后往前推,因为最后的人总能得到他想要的位置,那么从最后一个人开始插入到线段树中,找出每个人的真正位置。 sum[rt]代表着这个节点下还有多少个空位。

#include<cstdio>
#define lson l , m ,rt << 1
#define rson m+1 , r , rt << 1 | 1
const int maxn = 200002;
int sum[maxn << 2], num[maxn];
struct point
{
    int u,v;
}p[maxn];

void build(int l,int r,int rt)
{
    sum[rt] = r - l + 1 ;
    if(l==r)  return;
    int m=(l+r) >> 1;
    build(lson);
    build(rson);
}

int query(int p,int l,int r,int rt)
{
    sum[rt]--;   //空位数减 1 
    if(l==r) return l;
    int m = (l+r) >> 1;
    if(sum[rt<<1] >= p) return query(p , lson);
    else return query(p-sum[rt<<1] , rson); //查询右子树  
}

int main()
{
    int n,i;
    while(~scanf("%d",&n))
    {
        build(1,n,1);
        for(i=0;i<n;i++)
        {
            scanf("%d%d",&p[i].u,&p[i].v);
        }
        for(i=n-1;i>=0;i--)
        {
            num[query(p[i].u+1,1,n,1)]=p[i].v;  //query第i个数的真正位置
        }
        for(i=1;i<n;i++)
        {
            printf("%d ",num[i]);
        }
        printf("%d\n",num[n]);
    }
    return 0;
}