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-7struct 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;}