pku2828 Buy Tickets

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

线段树，单点更新

反向添加，每次放到第a[i]个空位上，用线段树维护区间内空位数，

单点查询，查询第x个空位的下标；

#include <stdio.h>

#define lson l, m, root<<1
#define rson m+1, r, root<<1|1

const int maxn = 200010;
int sum[maxn<<2];

void push_up(int root)
{
    sum[root] = sum[root<<1] + sum[root<<1|1];
}

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

void update(int x, int l, int r, int root)
{
    int m;
    if(l == r)
    {
        sum[root] = 0;
        return;
    }
    m = (l + r) >> 1;
    if(x <= m)
    {
        update(x, lson);
    }
    if(m+1 <= x)
    {
        update(x, rson);
    }
    push_up(root);
}

int query(int x, int l, int r, int root)
{
    int m;
    if(l == r)
    {
        return l;
    }
    m = (l + r) >> 1;
    if(x <= sum[root<<1])
    {
        return query(x, lson);
    }
    return query(x-sum[root<<1], rson);
}

int main()
{
    int i, n, a[maxn], b[maxn], result[maxn], temp;
    while(~scanf("%d", &n))
    {
        for(i=1; i<=n; i++)
        {
            scanf("%d%d", a+i, b+i);
            a[i] ++;
        }
        build(1, n, 1);
        for(i=n; i>=1; i--)
        {
            temp = query(a[i], 1, n, 1);
            result[temp] = i;
            update(temp, 1, n, 1);
        }
        for(i=1; i<=n; i++)
        {
            printf(i-n? "%d ": "%d\n", b[result[i]]);
        }
    }
    return 0;
}