HDU5493 Queue 线段树单点更新

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5493

题目大意：一个队列中有n个身高不同的人，现在他们吃饭回来要站回原来的队列，每个人只记得他前面或后面有k个比他高的人，现在给你每个人的身高，要还原出一个字典序最小的队列。

分析：和POJ2828 Buy Tickets有点相似，做法大体相同，我们先把这n个人按身高升序排序（定义结构体p存放地每个人的k和h），对于第i个人，如果他看到的比他高的人的人数大于实际比他高的人的人数（即p[i].k>n-i），那么显然是无法还原的；否则，他在队列中的位置就是第p[i].k+1（他前面有k个比他高的）或者n-i-p[i]+1（他后面有k个比他高的）个，由于要还原出字典序最小的一个，我们选择两者之间的较小值作为他在队列中的位置。实现上，我们可以用线段树来纪录每个区间的空位置数，用叶子节点来纪录队列中第i个人的身高。

实现代码如下：

#include <cstdio>
#include <algorithm>
using namespace std;
const int N=1e5+10;
struct person
{
int h,k;
bool operator <(const person &a) const
{
return h<a.h;
}
}p
;
struct segment
{
int l,r;
int emp;
int mid()
{
return (l+r)>>1;
}
}tree[N<<2];
int heigh
;
void build(int rt,int l,int r)
{
tree[rt].l=l;
tree[rt].r=r;
tree[rt].emp=r-l+1;
if(l==r) return ;
int m=tree[rt].mid();
build(rt<<1,l,m);
build(rt<<1|1,m+1,r);
}
void update(int rt,int l,int r,int h,int k)
{
if(tree[rt].l==tree[rt].r)
{
tree[rt].emp=0;
heigh[l]=h;
return ;
}
int m=tree[rt].mid();
if(k<=tree[rt<<1].emp) update(rt<<1,l,m,h,k);
else update(rt<<1|1,m+1,r,h,k-tree[rt<<1].emp);
tree[rt].emp=tree[rt<<1].emp+tree[rt<<1|1].emp;
}
int min(int a,int b)
{
return a<b?a:b;
}
int main()
{
int t,T=1,n;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
build(1,1,n);
for(int i=1;i<=n;i++)
scanf("%d%d",&p[i].h,&p[i].k);
sort(p+1,p+1+n);
bool flag=true;
for(int i=1;i<=n;i++)
{
if(p[i].k>n-i)
{
flag=false;
break;
}
int k=min(p[i].k,n-i-p[i].k);
update(1,1,n,p[i].h,k+1);
}
printf("Case #%d:",T++);
if(!flag)
{
puts(" impossible");
continue;
}
for(int i=1;i<=n;i++)
printf(" %d",heigh[i]);
printf("\n");
}
return 0;
}