NCSTOJ: [算法竞赛进阶指南]Lost Cows

List
Contest/Task
Wanna change your avatar? Read FAQ! User Portal
D : [算法竞赛进阶指南]Lost Cows
Time Limit:2 Sec Memory Limit:128 MiB
Back Submit Edit

Description
有n头奶牛，已知它们的身高为 1~n 且各不相同，但不知道每头奶牛的具体身高。

现在这n头奶牛站成一列，已知第i头牛前面有Ai头牛比它低，求每头奶牛的身高。

Input
第1行：输入整数n。

第2…n行：每行输入一个整数Ai,第i行表示第i头牛前面有Ai头牛比它低。

（注意：因为第1头牛前面没有牛，所以并没有将它列出）

Output
输出包含n行，每行输出一个整数表示牛的身高。1 ≤ n ≤ 10^5

第i行输出第i头牛的身高。

Sample Input
5
1
2
1
0
Sample Output
2
4
5
3
1

#include<iostream>
using namespace std;
const int maxn=1e5+10;
int a[maxn],b[maxn],c[maxn];
//三个数组分别存储原数据，树状数组和最终结果
//其中树状数组初始为全1数组的前序和，这样以便统计某一个数是否存在时
//本题的数据关系是：
//最后一个数据加一则表示它本身的序列值
//然后在树状数组中将其删除那么在判断倒数第二个数据时：
//在变化之后的树状数组中寻找第该元素加1个数即表示在剩下可选数据中
//它占到第几的位置
int n;
inline int lowbits(int x)
{
	return x&-x;
}
void insert(int x,int y)
{
	for(;x<=n;x+=lowbits(x))
		b[x]+=y;
}
int ask(int x)
{
	int  ans=0;
	for(;x;x-=lowbits(x))
		ans+=b[x];
	return ans;
}
int look(int q)
{//利用二分法在树状数组中寻找前序和为q的位置即当前需要寻找的排列顺序
	int l=1,r=n,mid=(l+r)/2;
  while(l<r)
  { mid=(l+r)/2;
  	if(ask(mid)>=q)r=mid;
  	else l=mid+1;
  }
  insert(l,-1);//将寻找的数据在前序和中删除
  return l;


}
int main(){
	scanf("%d",&n);
	a[1]=1;
     for(int i=2;i<=n;i++)
     {
     	scanf("%d",&a[i]);
     	a[i]++;//由题意得身高排序是从1开始的
     	//而原数据的大小关系是从0开始的
     	//整体加一不影响它们之间的顺序排列关系
     	//另外1位置也要从0数据变为1
     }

 for(int i=1;i<=n;i++)
 	insert(i,1);//初始化树状数组为全1数组的前序和

     for(int i=n;i>=1;i--)
     {
c[i]=look(a[i]);
     }
     for(int i=1;i<=n;i++)
     	printf("%d\n",c[i]);
     return 0;
     



}