AcWing 244. 谜一样的牛(树状数组+二分）

题目：https://www.acwing.com/problem/content/245/

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

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

输入格式

第1行：输入整数n。

第2..n行：每行输入一个整数Ai，第i行表示第i头牛前面有Ai头牛比它低。
（注意：因为第1头牛前面没有牛，所以并没有将它列出）

输出格式

输出包含n行，每行输出一个整数表示牛的身高。

第i行输出第i头牛的身高。

数据范围

1≤n≤1e5

输入样例：

5
1
2
1
0

输出样例：

2
4
5
3
1

题解：我们可以知道每次从后面的牛可以得到这头牛在前面的牛中身高排第几位，例如：An = 0，那第n头牛在前n头牛中是最高的，由此，我们可以用树状数组维护一个前缀和，从后面遍历（每次删除后面的牛），二分得到一个值为sum（i）的最小的位置，即为这头牛的身高。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e5+5;
int a[N],c[N],b[N];
int lowbit(int x) {
    return x&-x;
}
void add(int x,int y) {
    for(int i=x;i < N;i+=lowbit(i)) 
          c[i] += y;
}
int sum(int x) {
    int ans = 0;
    for(int i=x;i;i-=lowbit(i))
       ans += c[i];
    return ans;
}
int main() {
    int n;
    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>=1;i--) {
        int l = 1,r = n;
        while(l < r) {
            int mid = l + r >> 1;
            if(sum(mid) < a[i] + 1) l = mid + 1;
            else r=  mid;
        }
        add(l,-1);
        b[i] = l;
    }
    for(int i=1;i<=n;i++) cout << b[i] << endl;
}