244. 谜一样的牛

有 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

代码1：

/*
所以说,我们需要建立一个长度为n的1序列b,刚开始都是1,然后n到1倒序扫描每一个Ai,对于每个Ai执行查询和修改操作.
也就是说这道题目的题意就是让我们,动态维护一个01序列,支持查询第k个1所在的位置,以及修改序列中的一个数值
从遍历转换成 树状数组加二分  复杂度从O(n)->O(log2n)
*/
#include<bits/stdc++.h>
using namespace std;

#define x first
#define y second
#define endl '\n'
#define IOS ios::sync_with_stdio(false); cin.tie(0);cout.tie(0);
typedef long long LL;
typedef unsigned long long ull;
typedef pair<int, int> PII;

LL lowbit(LL x) { return x & (- x); }
LL gcd(LL a, LL b) { return b ? gcd(b, a % b) : a; }
const int mod = 1e9 + 7;
const int N = 100010;

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


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()
{
    IOS;
    cin >> n;

    for (int i = 2; i <= n; i++)
        cin >> h[i];

    // 可以就用add初始化 O(nlogn)
    for (int i = 1 ; i <= n ; i++)
        tr[i] = lowbit(i);//O(n)的初始化方法

    for (int i = n; i; i--)
    {
        int k = h[i] + 1;
        int l = 1 , r = n;

        while (l < r)
        {
            int mid = r + l >> 1;

            if (sum(mid) >= k) r = mid;
            else
                l = mid + 1;
        }

        ans[i] = r;
        add(r, -1);
    }

    for (int i = 1; i <= n ; i++)
        cout << ans[i] << endl;

    return 0;
}

代码2：

//vector 也能做
#include<bits/stdc++.h>
using namespace std;

#define x first
#define y second
#define endl '\n'
#define IOS ios::sync_with_stdio(false); cin.tie(0);cout.tie(0);
typedef long long LL;
typedef unsigned long long ull;
typedef pair<int, int> PII;

LL lowbit(LL x) { return x & (- x); }
LL gcd(LL a, LL b) { return b ? gcd(b, a % b) : a; }
const int mod = 1e9 + 7;
const int N = 1e6 + 10;
std::vector<int> q;
int n;
int h[N];
int ans[N];
int main()
{
    IOS;
    cin >> n;

    for (int i = 2; i <= n ; i++)
        cin >> h[i];

    for (int i = 1; i <= n; i++)
        q.push_back(i);

    for (int i = n ; i >= 1; i--)
    {
        int k = h[i] + 1;
        ans[i] = q[k - 1];
        q.erase(lower_bound(q.begin(), q.end(), q[k - 1]));
    }

    for (int i = 1; i <= n ; i++)
        cout << ans[i] << endl;

    return 0;
}