pku 2828 Buy Tickets（线段树）

倒着往队列里面插，每插入一个位置，就说明这个位置已经有人占了，删除这个位置。

比如说数据 pos value,那么就寻找队列中除去已经有人占据的位置后的第pos+1个位置。

 

我的代码是用标准规格的线段树写的，节点有left，right，cnt（记录该区段内被占据的位置的数量）域。

看到有些别人的代码，直接只用节点记录该区段内剩余有效位置数量，把insert和search操作结合，是个很不错的简化方式。

 

#include <iostream> using namespace std; const int maxn=200005; struct Node { int left,right,cnt;//cnt记录该区段内已经被删除的位置的个数 }tree[maxn*8]; void Build(int i,int left,int right) { tree[i].left=left; tree[i].right=right; tree[i].cnt=0; if(right-left>0) { Build(i*2,left,(left+right)/2); Build(i*2+1,(left+right)/2+1,right); } } void Insert(int i,int k)//返回出去被删除的位置后的第k个位置 { tree[i].cnt++; if(tree[i].right==tree[i].left) return; if(k>(tree[i].right+tree[i].left)/2) Insert(i*2+1,k); else Insert(i*2,k); } int Query(int i,int k) { if(tree[i].cnt==0) return tree[i].left+k-1; int temp=tree[2*i].right-tree[2*i].left-tree[2*i].cnt+1; if(k<=temp) Query(2*i,k); else Query(2*i+1,k-temp); } int queue[maxn]; int pos[maxn],value[maxn]; int main() { int n; while(scanf("%d",&n)!=EOF) { Build(1,1,n); for(int i=0;i<n;i++) { scanf("%d%d",&pos[i],&value[i]); } for(int i=n-1;i>=0;i--) { int m=Query(1,pos[i]+1); queue[m]=i; Insert(1,m); } for(int i=1;i<n;i++) { printf("%d ",value[queue[i]]); } printf("%d/n",value[queue[n]]); } return 0; }