迷失的牛【poj2182】【树状数组】【01前缀二分】

题目传送门

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

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

题目的性质：1~n各不相同
倒序扫描：第k头牛前面有Ak头牛比它低，那么他的身高Hk是数值1~n中第Ak+1没有出现过的数，倒序扫描一遍标记一个答案

在这里插入代码片

我们可以把1~n都标记1；
从n倒序扫描每个数，我们需要俩个操作：
1 查询序列b中第Ai+1个1在什么位置，这个位置就是奶牛的身高；
2 把这个位置变为0；

在这里插入代码片

我们需要动态维护一个01序列，支持查询时二分答案通过询问ask（mid） 的前缀和，可以得到mid前有多少个1，等到答案

#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int N=1e5+9;
int a[N],h[N],b[N];
int n;
int lowbit(int x)
{
    return x&-x;
}

void add(int pos,int val)
{
    for(;pos<=n;pos+=lowbit(pos))
        b[pos]+=val;
}
int ask(int pos)
{
    int res=0;
    for(;pos;pos-=lowbit(pos))
        res+=b[pos];
    return res;
}

int main()
{
    ios::sync_with_stdio(false);
    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>0;i--)
    {
        int l=1,r=n;
        while(r>l)
        {
            int mid=l+r>>1;
            if(ask(mid)>=a[i]+1) r=mid;
            else l=mid+1;
        }
        add(l,-1);
        h[i]=l;
    }
    for(int i=1;i<=n;i++)
        cout<<h[i]<<endl;
    return 0;
}