【POJ】2828 Buy Tickets 线段树

【POJ】2828 Buy Tickets

---

首先建造一棵 根节点为[1-n] 的线段树， 最底层节点就表示队列的每个位置。
即, 最后的答案是把 val 都填到线段树的子节点 [1,1]、[2,2]、[3,3]……中；

---

区间 [l,r] 的标记储存 [l,r] 这个区间中的最底层的节点有多少是空的。
比如

tree[1].l = 1;
tree[1].r = n;
tree[1].sum = n;
//这是初始状态根节点的值
//表示在 [1,n] 内有 n 个空位；
tree[1].l = 1;
tree[1].r = n;
tree[1].sum = n-1;
//表示在 [1,n] 内有 n-1 个空位, 也就是说我们已经处理了一个人的位置。

绿色表示标记，括号表示区间。

为了方便输出和储存，我们不用线段树的最底层节点存储 val， 我们再开一个数组 ans[x] 表示 第 x 个位置的人的 val 是谁。
这样输出的时候就不用再遍历一遍线段树了。

---

读入的数据到倒叙进行处理。
理由：
靠后面的数更可能是最后的标准答案。
如：

//样例
4
0 77
1 51
1 33
2 69

可知 69一定排在队列的第 3 个位置（前面有2个人）

---

deal 函数：
思想，找到刚好大于 pos 位置的节点。
当 处理的这个人 pos 时， 如果此时线段数的左节点拥有的空位数大于等于 这个人需要的pos 即，需要前面有多少人时， 走线段数左儿子；
如果线段数左儿子所拥有的空位数比 此 pos 小时， 那就不能往左边放， 于是我们进去右儿子。
直至 到达最底层节点。

---

样例模拟 （引 作者： ZH奶酪——张贺 ）：

//至于 pos 为什么是3 而不是2 的原因：因为当 pos == 0 时，就代表他是队首的那一个，即第一个。 以此类推.

插入3 69
1，4结点有4个位置，
1，2结点有2个位置，小于3，因此放到[1，4]结点右孩子，且[1，4]结点空位置减1
到了1，4右孩子后，只要找到第3-2=1个位置即可，而3，4结点的左孩子3，3含有1个空位置，1>=1，所以放到3，3位置了。

插入2 33

插入2 51
此时1，4的左孩子只有1个位置，1<2，所以只能放到1，4的右孩子3，4上
3，4的左孩子有0个位置，所以只能放在3，4的右孩子4，4上。

插入1 77

---

代码：

#include <iostream>
#include <queue>
#include <cstdio>
#include <cstring>
#define L(x) (x << 1)
#define R(x) (x << 1 | 1)
using namespace std;

const int MAXN = 200000 + 20;
int n;
int ans[MAXN];

struct TREES{
    int l, r, sum;
}tree[MAXN << 2];

struct Date{
    int p, v;
}d[MAXN];

void updata(int p)
{
    tree[p].sum = tree[L(p)].sum + tree[R(p)].sum;
}

void build(int l, int r, int p)
{
    tree[p].l = l;
    tree[p].r = r;
    tree[p].sum = r-l+1;
    if(l == r)
        return;
    int mid = (tree[p].l + tree[p].r)/2;
    build(l,mid,L(p));
    build(mid+1,r,R(p));
    return;
}

void deal(int p, int pos, int val)
{
    if(tree[p].l == tree[p].r)
    {
        tree[p].sum = 0;
        ans[tree[p].l] = val;
        return;
    }
    if(tree[L(p)].sum >= pos)
        deal(L(p), pos, val);
    else deal(R(p), pos-tree[p*2].sum, val);
    updata(p);
}

void init()
{
    for(int i = 1; i <= n; i ++)
        tree[i].l = tree[i].r = tree[i].sum = d[i].p = d[i].v = 0;
}

int main()
{
    while(scanf("%d",&n) != EOF)
    {
        init();
        build(1,n,1);
        for(int i = 1; i <= n; i ++)
            scanf("%d %d", &d[i].p, &d[i].v);
        for(int i = n; i >= 1; i --)
            deal(1, d[i].p+1, d[i].v);
        for(int i = 1; i < n; i ++)
            printf("%d ", ans[i]);
        printf("%d\n",ans[n]);
    }
    return 0;
}