POJ - 2828 Buy Tickets(线段树)

把插队的顺序从后往前进行query，每次找到第p[i]个不为空的位置，然后把p[i]位置设为空，这就变成单点修改，区间求和的线段树。

#include <iostream>
#include <cstdio>
#include <cstdlib>
using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int maxn=220000;
int sum[maxn<<2], pos[maxn], p[maxn], val[maxn];

void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l, int r, int rt)
{
    if(l==r)
    {
        sum[rt]=1;
        return ;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void query(int p, int v, int l, int r, int rt)
{
    if(l==r)
    {
        pos[l]=v;
        sum[rt]=0;
        return ;
    }
    int mid=(l+r)>>1;
    if(sum[rt<<1]>p)
        query(p, v, lson);
    else query(p-sum[rt<<1], v, rson);
    pushup(rt);
}

int main()
{
    int n;
    while(~scanf("%d", &n))
    {
        build(0, n-1, 1);
        for(int i=0; i<n; i++)
            scanf("%d%d", &p[i], &val[i]);
        for(int i=n-1; i>=0; i--)
            query(p[i], val[i], 0, n-1, 1);
        for(int i=0; i<n; i++)
            printf("%d%c", pos[i], i==n-1?'\n':' ');
    }
    return 0;
}