POJ 2828:Buy Tickets

来源:互联网 发布:动漫壁纸网站 知乎 编辑:程序博客网 时间:2024/06/05 03:43

点击打开题目链接

题目翻译:

就是往一个队列里面插队,给出一个数n,给出n 个 pos 和 value。将value插入到第pos个人的右侧,如果这个位置已经有人了,则

此人占据此位置,后面的人依次后挪一位。

解题思路:构建初始构建空的线段树,开辟sum数组,用来记录相应区间的空位个数。由于正着插如果某个位置上有人了,则后面

位置上的人要向右挪。这样去考虑题目很复杂,因此可以想着逆序将人插入线段树,因为最后一个进入某个位置的人必然占据某位置,

然后在往有人的位置上插人时,这个人往后挪插到别处。如果将一个人插入到第pos个人的右侧,则其前方必然有pos个空位置,它自

己也要占据空位置,则总共有pos+1个空位置,在线段树中进行查找,如果左子树所有的空位置,满足大于等于pos+1.则插入到左子

树中,否则的话,首先左侧已经有sum[root<<1]个空位置了,只需在右子树中查pos+1-sum[root<<1]个空位就行了,当查到叶子节点

的时候,把val值插入,叶子节点的sum值更新,层层递归更新上层节点的空位置的值。

代码比较简洁:POJ上难得的1A,因为原来做题总是被坑,错好多次才过。这个题一次提交就过了。


#include <iostream>#include <stdio.h>#define lchild left,mid,root<<1#define rchild mid+1,right,root<<1|1using namespace std;const int maxn = 200010;int sum[maxn<<2],ans[maxn],position[maxn],value[maxn];void push_up(int root){    sum[root] = sum[root<<1] + sum[root<<1|1];}//构建线段树void build(int left,int right,int root){    sum[root] = right-left+1;    if(left == right) return;    int mid = (left + right)>>1;    build(lchild);    build(rchild);}void insert(int pos,int left,int right,int root,int val){    if(left == right)    {        sum[root] = 0;        ans[left] = val;        return;    }       int mid = (left + right)>>1;    if(pos<=sum[root<<1]) insert(pos,lchild,val);    else insert(pos-sum[root<<1],rchild,val);    push_up(root);}int main(){    int n;    while(~scanf("%d",&n))    {        for(int i = 1; i <= n; i++)        {            scanf("%d%d",&position[i],&value[i]);        }        build(1,n,1);        for(int i = n; i > 0; i--)        {                insert(position[i]+1,1,n,1,value[i]);        }        printf("%d",ans[1]);        for(int i = 2; i <= n; i++)        {                printf(" %d",ans[i]);        }        printf("\n");    }    return 0;}