244. 谜一样的牛

有 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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值