树状数组——谜一样的牛

传送门：244. 谜一样的牛 - AcWing题库

思路：题意转化一下就是，建立一个初始为1的01序列c，从后往前扫，每一头牛的身高就是在序列c中第a[i]+1个1出现的位置下标。

维护一个01序列，支持查询第k个1的位置，以及修改序列中的一个数值。

这里可以用二分查找第k个1，因为是在前缀和数组中查找k的值，满足非降序，所以是可以使用二分查找。

拿测试样例过一遍：

 

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<stack>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int c[N];
int a[N];
int ans[N];
int n;
int lowbit(int x)//作用：得到x的二进制下最小的2的次幂
{
    return x&(-x);
}
void add(int x,int y)
{
    for(;x<=n;x+=lowbit(x)) c[x]+=y;
}
int ask(int x)
{
    int ans=0;
    for(;x;x-=lowbit(x)) ans+=c[x];
    return ans;
}
int main()
{
    scanf("%d",&n);
    add(1,1);
    for(int i=2;i<=n;i++)
       {
        scanf("%d",&a[i]);
       add(i,1);
       }
    for(int i=n;i>=1;i--)
    {
        int l=1,r=n;
        while(l<r)
        {
            int mid=l+r>>1;
            if(ask(mid)<a[i]+1)  l=mid+1;
            else r=mid;
        }
        ans[i]=r;
        add(r,-1);
    }
    for(int i=1;i<=n;i++)
    cout<<ans[i]<<endl;
    return 0;
}