AcWing 244. 谜一样的牛
原题链接
简单
作者:
Anoxia_3
,
2020-07-21 14:29:58
,
所有人可见
,
阅读 465
/*
找的顺序是从后往前依次遍历每一头牛,每头牛都是第h[i]+1低,于是在剩下的身高中找第h[i]+1小的值,再删除该数即可
于是需要支持的操作就是:
①在剩下的数中找到第x小的值,二分,判断依据利用树状数组中的sum(t),即求出1~t中可用的身高是否大于等于x
②删除第x小的数,树状数组中的add操作,如果某个身高被用过-1即可。
②就是;
②可以利用
*/
#include <iostream>
using namespace std;
const int N = 100010;
int n;
int h[N] , tr[N];
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;
}
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 >> h[i];
for(int i = 1 ; i <= n ; i++) add(i , 1);
for(int i = n ; i ; i--)
{
int l = 1 , r = n;
while(l < r)
{
int mid = l + r >> 1;
if(sum(mid) >= h[i] + 1) r = mid;
else l = mid + 1;
}
ans[i] = l;
add(l , -1);
}
for(int i = 1 ; i <= n ; i++) cout << ans[i] << endl;
return 0;
}