有 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
分析
- 通过样例分析,不能直接确定第一个位置的牛多高,但是可以确定最后一个位置的牛是多高,a[n]=0,k=a[n]+1,说明前面有0头牛比他低,那么最后一头牛的高度就是第k小的数(1~n中未被使用的顺序);
- 所以可以使用树状数组维护01序列的前缀和,便于查询k所在位置;从剩余的数中找第k小的数,可以转化为前缀和sum(x)=k这个条件,那么就转化为二分去查找这个满足条件的sum(x);从[1,n]二分x(x就是mid),找到满足条件的x保存在t,然后存进方案ans;记得单点修改t处的值,减1变成0即可;
- 在确定牛的身高,是倒着去确定的;并且记得初始化树状数组,因为维护的是01序列,且每个数均为被使用,可以直接初始化为1,而且此题的x也是表示位置,树状数组维护的都是前缀和,所以x一般都是索引;
#include <bits/stdc++.h>using namespace std;
typedef long long LL;
const int N = 100010;int n;
int a[N];
int tr[N];//维护01序列的前缀和,便于查询k所在位置
int ans[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;}
}//区间[1,x]的01序列和
int sum(int x) {int res = 0;for (int i = x; i; i -= lowbit(i)) {res += tr[i];}return res;
}int main() {cin >> n;for (int i = 2; i <= n; ++i) {cin >> a[i];}for (int i = 1; i <= n; ++i) {//初始化刚开始 1~n的数都没用过,都是1add(i, 1);}//倒着去确定for (int i = n; i; --i) {//找没用过的数中,第k小的数int k = a[i] + 1;int l = 1, r = n;int t;//第k小的数所在位置while (l <= r) {int mid = l + (r - l) / 2;if (sum(mid) >= k) {t = mid;r = mid - 1;} else {l = mid + 1;}}ans[i] = t;//第t个位置的数-1,变成了0,代表已用过add(t, -1);}for (int i = 1; i <= n; ++i) {cout << ans[i] << endl;}return 0;
}