POJ 2828 Buy Tickets(线段树+单点更新)


       题意是要排队买票,每个人都有一个它所排的位置,假如说第一个人在第一个位置,他就站在第一个位置,如果第二个人也要站在第一个位置的话,就会把第一个人挤到第二个位置上,如果第二个位置上有人的话,就会排到第三个位置上,以此类推,然后按顺序从前到后输出每个人的权值。

       思路就是倒着去排位置,因为最后一个插到这个位置的人是固定的,他是不会被挤走的,比如最后一个人在第2个位置上,那么他就一定会在第二个位置上,所以我们倒着去更新线段树就好了,线段树的话,就是每个结点记录当前区间有多少个空位,然后根据遍历的排队位置去判断,如果这个人的位置是4的话,左子结点的空位置为3的话,那只能去右子结点去找了,结合代码想想吧,应该挺好理解的。


AC代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#define lson l, mid, o << 1
#define rson mid + 1, r, o << 1 | 1
#define maxn 200005
using namespace std;
int pre[maxn << 2];
int ans[maxn];
struct Node{
  int x,y;
}Edge[maxn];
int n,flag;

void Pushup(int o){
  pre[o] = pre[o << 1] + pre[o << 1 | 1];
}

void Build(int l,int r,int o){
  if(l == r){
    pre[o] = 1;
    return ;
  }
  int mid = (l + r) >> 1;
  Build(lson);
  Build(rson);
  Pushup(o);
}

void Update(int x,int l,int r,int o){
  pre[o]--;
  if(l == r){
    flag = l;
    return ;
  }
  int mid = (l + r) >> 1;
  if( x <= pre[o << 1])Update(x,lson);
  else{
    x -= pre[o << 1];
    Update(x,rson);
  }
}

int main()
{
  while(~scanf("%d",&n)){
    Build(1,n,1);
    for(int i=0;i<n;i++){
      scanf("%d%d",&Edge[i].x,&Edge[i].y);
    }
    for(int i=n-1;i>=0;i--){
      Update(Edge[i].x + 1, 1, n, 1);
      ans[flag] = Edge[i].y;
    }
    for(int i=1;i<=n;i++){
      printf("%d%c",ans[i],i==n?'\n':' ');
    }
  }
  return 0;
}

猜你喜欢

转载自blog.csdn.net/charles_zaqdt/article/details/80572540