python/TangDou/AcWing/BIT/244_ErFen_1.cpp

#include <bits/stdc++.h>
using namespace std;

const int N = 100010;
int a[N];
int ans[N];
int n;

// 树状数组模板
typedef long long LL;
#define lowbit(x) (x & -x)
int c[N];
void add(int x, int d) {
    for (int i = x; i < N; i += lowbit(i)) c[i] += d;
}
LL sum(int x) {
    LL res = 0;
    for (int i = x; i; i -= lowbit(i)) res += c[i];
    return res;
}

int main() {
    scanf("%d", &n);
    for (int i = 2; i <= n; i++) scanf("%d", &a[i]); // 读入每头奶牛前面有几头奶牛比自己矮
    // 利用树状数组，把1~n之间每个位置设置为1,方便后面用来求前缀和
    for (int i = 1; i <= n; i++) add(i, 1);

    for (int i = n; i; i--) {
        int l = 1, r = n;
        while (l < r) {             // 用二分找出前缀和恰好为a[i] + 1的数
            int mid = (l + r) >> 1; // mid枚举的是下标
            if (sum(mid) >= a[i] + 1)
                r = mid; // 再小点试试
            else
                l = mid + 1; // 再大点~
        }
        ans[i] = l; // 记录答案
        add(l, -1); // 找到后，这个数-1,标识为0，方便下次求前缀和
    }
    // 输出结果
    for (int i = 1; i <= n; i++) printf("%d\n", ans[i]);
    return 0;
}