谜一样的牛 - 题解

题目描述

有n头奶牛，已知它们的身高为 1~n 且各不相同，但不知道每头奶牛的具体身高。
现在这n头奶牛站成一列，已知第i头牛前面有Ai头牛比它低，求每头奶牛的身高。

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

输出格式
输出包含n行，每行输出一个整数表示牛的身高。
第i行输出第i头牛的身高。

样例

样例输入
5
1
2
1
0
样例输出
2
4
5
3
1

算法分析

这道题首先的思路肯定是暴力，但是老司机 有经验的人都知道暴力会直接超时，所有要其他方法。
其实这道题有很多方法，这里只讲二分+树状数组。
我们用一个树状数组，表示前$i$个数有多少个数是未被使用过的。然后，我们再用二分求出第$a[i]+1$大的数。
但如果$i$这个数被使用了，则要修改树状数组。这其实就是树状数组中最