谜一样的牛 - 树状数组二分

有 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

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'

using namespace std;

typedef pair<int, int> PII;
typedef long long ll;

const int N = 100010;

int n;
int h[N], tr[N], ans[N];

inline int lowbit(int x)
{
    return x & -x;
}

inline void add(int x, int c)
{
    for(int i = x; i <= n; i += lowbit(i))
    {
        tr[i] += c;
    }
}

inline int sum(int x)
{
    int res = 0;
    for(int i = x; i; i -= lowbit(i))
    {
        res += tr[i];
    }
    return res;
}

int main()
{
    IOS
    cin >> n;
    for(int i = 2; i <= n; i ++)cin >> h[i];
    
    for(int i = 1; i <= n; i ++)tr[i] = lowbit(i);
    
    for(int i = n; i; i --)
    {
        int x = h[i];
        int l = 1, r = n;
        while(l < r)
        {
            int mid = l + r >> 1;
            if(sum(mid) > x)r = mid;
            else l = mid + 1;
        }
        ans[i] = l;
        add(r, -1);
    }
    
    for(int i = 1; i <= n; i ++)cout << ans[i] << endl;
    
    return 0;
}

 倒着往前找就可以了，加入当前数为x，只要找到剩余数中第x + 1小的数就行了。

每个数都先标记为1，被用过之后就在树状数组中对这个数-1，然后就二分出第一个比x大的数了，

二分出的数就是奶牛的高度。

学自：Acwing