【POJ】2828 Buy Tickets 线段树

来源:互联网 发布:最完美的丁丁长度 知乎 编辑:程序博客网 时间:2024/06/07 01:56

【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 是谁。
这样输出的时候就不用再遍历一遍线段树了。


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

//样例40 771 511 332 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;}
2 0