ZOJ 3635 Cinema in Akiba (树状数组+二分)

//460ms
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define low(x) x&(-x)
int c[100005],ans[100005],n;
void add(int i,int val)
{
while(i<=n)
{
c[i]+=val;
i+=low(i);
}
}
int getsum(int i)
{
int sum=0;
while(i>0)
{
sum+=c[i];
i-=low(i);
}
return sum;//getsum(i)表示该位置前（包括）还有多少空位
}
int main()
{
int i;
while(scanf("%d",&n)!=EOF)
{
memset(c,0,sizeof(c));
memset(ans,0,sizeof(ans));
for(i=1;i<=n;i++)add(i,1);
for(i=1;i<=n;i++)
{
int mid,p;
scanf("%d",&p);
int r=n,l=1;
while(r>l)
{
mid=(r+l)/2;
//if(i==5)printf("===%d\n",mid);
if(getsum(mid)>=p)r=mid;//有重复值时向前找第一个
else l=mid+1;
}
ans[i]=r;
add(r,-1);
/*
一开始不是这样写的，wa到死!
错误代码：
while(l<=h)
{
int mid=(l+h)/2;
printf("mid=%d\n",mid);
if(p<getsum(mid)){
h=mid-1;
}
else if(p>getsum(mid))l=mid+1;
else
{
ans[i]=mid;
add(mid,-1);
// printf("--%d\n",mid);
break;
}
}
*/
}
int m;
scanf("%d",&m);;
while(m--)
{
int x;
scanf("%d",&x);
if(m)printf("%d ",ans[x]);
else printf("%d\n",ans[x]);
}

}
return 0;
}