谜一样的牛【树状数组 简易思路证明，一看就悟】

😊😊 😊😊
不求点赞，只求耐心看完，指出您的疑惑和写的不好的地方，谢谢您。本人会及时更正感谢。希望看完后能帮助您理解算法的本质
😊😊 😊😊

题目描述：

题目描述
有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. 初始，每头牛的前面有多少头比它矮的牛（序列）。
2. 由该序列算出它应该是第几小的牛，即排名至少是第几名。比如：a[5] = 3；表示前面有3头牛比第五只牛低，则第五只牛至少是第四名；
3. 唯一确定第五只牛是第四名的前提是第五只牛是当前未确定身高的牛中，排在末尾的那头牛。
4. 所以说：从后往前思考！！利用好 $a$ 数组；从而推出一个关键点：从后往前递推，处于序列的末尾的第 $i$ 头牛的身高，在身高序列 $(1\sim n)$ 至少是 第 $a[i]+1$ 名。
5. $n$只牛的身高序列是：$1\sim n$，对应于每只牛！所以牛的身高不可能重复。所以从后往前每次确定一只牛的身高，就要从当前序列（$1\sim n$）中删除该身高值
6. 接下来问题就是如何快速确定第 $i$ 只牛在当前序列中的身高呢？由于 第4点 说了：当前牛的身高排名应该是剩余的 $1\sim n$序列中，至少排：$(a[i]+1)$ 名的身高；所以说我们通过寻找 $1\sim n$中排名为 $a[i]+1$ 名的身高即可。那么如何确定当前数字在 $1\sim n$ 中的排名？将身高序列：$1\sim n$设为 01序列，即 $h[1]=1,h[2]=1,h[3]=\mathrm{1....}h[i]=1；$数组元素的下标，表示该身高是多少，取值为1表示该身高还存在于序列。反之为0，则表示不存在了！
   图例：
   
   显而易见，求第几名，就是要在 $1\sim n，即h数组$ 中求一个前缀和为 $(a[i]+1)$ 的下标；(h数组中下标就是身高)。
7. 因为我们每次删除一个身高后，还需要去动态地维护当前序列，对于删除的身高进行标记，并且还要通过前缀和去判断下一只牛的身高。所以 删除某个身高（单点更新）和 删除后还得更新序列求删除后的前缀和（动态维护区间和）；这就是树状数组的特性啦！
8. 另外值得注意的是：我们要找的是：序列 $1\sim n$ 前缀和至少为：$a[i]+1$ 的下标，假设：$a[i]+1=k$，则我们要找的是前缀和至少为 $k$的下标，即前缀和大于等于 $k$ 的第一个数，看到这，是不是不禁想到了二分查找，又由于 $h[]$序列的值为0或1，所以从左往右，区间长度越长，前缀和越大，符合单调性，所以可以采用二分查找。如果采用顺序枚举的话，你得从左往右枚举每一个下标，每次枚举一个标，就得求一次它的前缀和。时间复杂度较高！

代码：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int N = 1e5 + 10;
int a[N], c[N];     //原数组，树状数组！
int n;
int ans[N];

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

void add (int pos, int v)     //单点修改
{
    for (int i=pos; i<=n; i += lowbit(i))
        c[i] += v;
}

int sum(int pos)       //区间查询！
{
    int res=0;
    for (int i=pos; i>0; i-=lowbit(i))
        res += c[i];
    return res;
}

int main()
{
    scanf ("%d", &n);
    
    for (int i=2; i <= n; i ++)
        scanf ("%d", &a[i]);
    for (int i=1; i <= n; i ++)
        c[i] += lowbit(i);
    
    for (int i=n; i; i --)
    {
        int k = a[i] + 1;
        int l = 1, r = n;
        while (l < r)
        {
            int mid = (l + r) >> 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;
}