题目描述
有n头奶牛,已知它们的身高为 1~n 且各不相同,但不知道每头奶牛的具体身高。
现在这n头奶牛站成一列,已知第i头牛前面有Ai头牛比它低,求每头奶牛的身高。
输入格式
第1行:输入整数n。
第2…n行:每行输入一个整数Ai,第i行表示第i头牛前面有Ai头牛比它低。
(注意:因为第1头牛前面没有牛,所以并没有将它列出)
输出格式
输出包含n行,每行输出一个整数表示牛的身高。
第i行输出第i头牛的身高。
数据范围
1≤n≤105
输入样例:
5
1
2
1
0
输出样例:
2
4
5
3
1
题目分析
有n头牛,a[i]表示第i头牛前面有多少头牛比它矮。因为每一头牛只能看前面的,因此我们应该从最后一头牛开始往前算。
第i头牛前面有a[i]头牛比它矮,那么第i头牛的身高即为前i头牛中的第a[i]+1名。
我们可以用树状数组来维护一个初始为b[1]=b[2]=b[3]=…=b[n]=1的集合。b[i]=1表示身高为i的牛还未确定。然后从后往前枚举a[i],a[i]+1即为当前牛在剩余的这些牛中的身高为第a[i]+1名。这个第a[i]+1名可以用二分来算出来,最后再将这个身高存在ans[i]中即可。
代码如下
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <stack>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#include <iomanip>
#define LL long long
#define PII pair<int,int>
using namespace std;
const int N=1e5+5;
int n;
int a[N],tr[N];
int ans[N];
int lowbit(int x) //树状数组的三个基本函数
{
return x& -x;
}
void add(int x,int c)
{
for(int i=x;i<=n;i+=lowbit(i))
tr[i]+=c;
}
int sum(int x)
{
int res=0;
for(int i=x;i;i-=lowbit(i))
res+=tr[i];
return res;
}
int main()
{
scanf("%d",&n);
for(int i=2;i<=n;i++) //注意:输入是从第二头牛开始的
scanf("%d",&a[i]);
for(int i=1;i<=n;i++) add(i,1); //建立树状数组
for(int i=n;i>0;i--)
{
int l=1,r=n,k=a[i]+1;
while(r>l)
{
int mid=l+r>>1;
if(sum(mid)>=k) r=mid; //找到树状数组中大于等于k最小的数的位置
else l=mid+1;
}
ans[i]=r; //记录该位置
add(r,-1); //将该位置的数删除(因为该身高对应的牛已经确定了)
}
for(int i=1;i<=n;i++)
printf("%d\n",ans[i]);
return 0;
}