有 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
分析:二分+树状数组,吧所有点初始化为1,然后倒着遍历每个牛,若该牛占了第i个位置,则把第i个位置置为0;
代码如下:
#include <iostream>
using namespace std;
const int N=1e5+10;
int a[N],t[N],ans[N];
int n;
void add(int x,int w)
{
for( ; x<=n;x+=(x&-x))t[x]+=w;
}
int getsum(int x)
{
int res=0;
for(;x;x-=(x&-x))res+=t[x];
return res;
}
int main()
{
cin>>n;
for(int i=2;i<=n;i++)cin>>a[i];
for(int i=1;i<=n;i++)add(i,1);
for(int i=n;i;i--)
{
int r=n,l=1;
int te=a[i]+1;
while(l<r)
{
int mid=r+l>>1;
int tt=getsum(mid);
if(tt<te)l=mid+1;
else r=mid;
}
add(l,-1);
ans[i]=l;
}
for(int i=1;i<=n;i++)cout<<ans[i]<<endl;
return 0;
}