POJ 2828 Buy Tickets（单点更新）

题目链接:http://poj.org/problem?id=2828

对于第i个人的pos，可以看成在第i个人之前需要留的空位置数。
利用线段树可以确定每个人的最终位置。
那么对于每个人做一次查询，但需要 从后往前 查询。

代码如下。

#include <iostream>
#include <iomanip>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>
#include <stack>
#include <string>
#include <cstring>
#include <cassert>
using namespace std;

typedef long long ll;
const int maxn=222222;
const int INF=0x7fffffff;
const int mod=1e7+7;
#define LSON l, m, rt<<1
#define RSON m+1, r, rt<<1|1
#define ESP 1e-7

struct people {
    int pos, val;
}peo[maxn];

int arr[maxn<<2], put[maxn];

void pushup(int rt) {
    arr[rt]=arr[rt<<1]+arr[rt<<1|1];//每个节点存该区间剩余位置数
}

void build(int l, int r, int rt) {
    if(l==r) {
        arr[rt]=1;
        return ;
    }
    int m=(l+r)>>1;
    build(LSON);
    build(RSON);
    pushup(rt);
}

int query(int pos, int l, int r, int rt) {
    if(l==r) {
        arr[rt]--;
        return l;
    }
    int m=(l+r)>>1;
    int temp=pos<=arr[rt<<1]?query(pos, LSON):query(pos-arr[rt<<1], RSON);
    pushup(rt);//查询同时更新
    return temp;
}

int main() {
    int n;
    while(~scanf("%d", &n)) {
        build(1, n, 1);
        for(int i=0;i<n;i++) {
            scanf("%d%d", &peo[i].pos, &peo[i].val);
            peo[i].pos++;
        }
        for(int i=n-1;i>=0;i--)//逆序
            put[query(peo[i].pos, 1, n, 1)]=peo[i].val;
        for(int i=1;i<n;i++)
            printf("%d ", put[i]);
        printf("%d\n", put[n]);
    }
    return 0;
}