poj 2828 Buy Tickets (排队问题+线段树)

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 200005
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int sum[maxn<<2],a,b;
struct qw{
int a,val,num;
}an[maxn];
int cmp(struct qw x,struct qw y){
return x.num<y.num;
}
void push_up(int rt){
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt){
if(l==r){
sum[rt]=1; return;
}
int m=(l+r)>>1;
build(lson); build(rson);
push_up(rt);
}
void update(int p,int l,int r,int rt){
if(l==r){
sum[rt]=0; return;
}
int m=(l+r)>>1;
if(p<=m) update(p,lson);
else update(p,rson);
push_up(rt);
}
int q;
//2分查找q,使得sum(1,q)==p; i就是此点要插入的位置
void bina_query(int p,int l,int r,int rt){
if(l==r&&p==sum[rt]){ //刚开始少了(l==r)这个条件,由于必须是肯定到某点以后恰好为p,所以不加会错
q=r; return;
}
int m=(l+r)>>1;
if(p<=sum[rt<<1]) bina_query(p,lson);
else bina_query(p-sum[rt<<1],rson);
}
int main(){
int i,j,n;
while(~scanf("%d",&n)){
build(1,n,1);
for(i=1;i<=n;i++) scanf("%d%d",&an[i].a,&an[i].val);
//倒着插入
for(i=n;i>=1;i–){
bina_query(an[i].a+1,1,n,1);
an[i].num=q;
update(q,1,n,1);
}
sort(an+1,an+1+n,cmp);
for(i=1;i<n;i++) printf("%d ",an[i].val);
printf("%d
",an[i].val);
}
return 0;
}