树状数组+二分：谜一样的牛

原题链接：https://www.acwing.com/activity/content/code/content/164773/

树状数组+二分

特殊性质：
刚开始c数组都为1。 从最后一个数开始，如果前边有k头牛比他矮，那他就是第k+1为1的数，也就是k+1, 同时标记k+1 被用过了。c[k+1] = 0; 然后倒数第二个数，如果前边有j头牛比他矮。那他就是c数组中第j+1个为1的数。同时标记对应位置为0。

c[1~n] 都初始化成1， 然后代表1~n都还没用过。然后依次从后往前枚举h[i] 如果用k过了，就在c中对应c[k]标记为0，同时维护树状数组t。

对于每个位置i，如何快速找到可用的第h[i]+1个数。由于用过的在c[]中已经为0了。没用过的，还是1，直接树状数组求前缀和。在1-n的区间内，找到sum(j) >= h[i]+1 的最小的数

也就是说这道题目的题意就是让我们,动态维护一个01序列,支持查询第k个1所在的位置,以及修改序列中的一个数值

如果不使用树状数组的话，那么修改为O(1), 但是查询是O(n), 总的时间复杂度就是O(n^2), 会超时。
现在变成了查询是O(logn * logn), 修改时O(logn), 总的是O(nlogn * logn)

/*
    树状数组+二分
    
    特殊性质：
    刚开始c数组都为1。从最后一个数开始，如果前边有k头牛比他矮，那他就是第
    k+1为1的数，也就是k+1, 同时标记k+1 被用过了。c[k+1] = 0; 
    然后倒数第二个数，如果前边有j头牛比他矮
    那他就是c数组中第j+1个为1的数。同时标记对应位置为0
    
    c[1~n] 都初始化成1， 然后代表1~n都还没用过。然后依次从后往前枚举h[i]
    如果用k过了，就在c中对应c[k]标记为0，同时维护树状数组t。
    
    对于每个位置i，如何快速找到可用的第h[i]+1个数。由于用过的在c[]中已经为0了
    没用过的，还是1，直接树状数组求前缀和。在1-n的区间内，找到sum(j) >= h[i]+1 
    的最小的数
    
    也就是说这道题目的题意就是让我们,动态维护一个01序列,支持查询
    第k个1所在的位置,以及修改序列中的一个数值
*/
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n;
int h[N];
int ans[N];
// 假设原数组为c，c的树状数组为tr
int tr[N];

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

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

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

int main()
{
    scanf("%d", &n);
    // 从2开始读
    for (int i = 2; i <= n; i ++ ) scanf("%d", &h[i]);

    // 初始化c，同时维护tr
    for (int i = 1; i <= n; i ++ ) 
    {
            // tr[i] = lowbit(i);
            add(i, 1);  // 这俩操作是等价的
    }

    // 倒着来
    for (int i = n; i; i -- )
    {
        // 前边有h[i]个比自己小的，那自己就h[i]+1个
        int k = h[i] + 1;
        int l = 1, r = n;
        while (l < r)
        {
            int mid = l + r >> 1;
            // 找到sum(j) >= h[i]+1 的最小的数
            if (sum(mid) >= k) r = mid;
            else l = mid + 1;
        }
        // 找到r之后，存到ans中
        ans[i] = r;
        // 同时修改c[r]为0，维护树状数组t。
        add(r, -1);
    }

    for (int i = 1; i <= n; i ++ ) printf("%d\n", ans[i]);

    return 0;
}