谜一样的牛【树状数组+二分】
POJ2182
、ACwing244
思路:
我们可以用树状数组维护一个bool数组,在1~n如果某个位置是1说明现在这个高度的牛还未出现,
而如果是0,说明已经出现了。
例如 1 1 1 1
从第n头牛开始倒序遍历:
如果第4头牛前面有1头牛比它低,那么它的身高应该是2(bool数组的第二个1出现的位置)
此时:1 0 1 1
第n-1头牛前面有1有牛比它低,那么它的身高应该是3(bool数组的第二个1出现的位置)
那么我们可以使用一个树状数组来维护,以logn的时间对前缀和进行二分查找,找到目标位置,
而每次利用树状数组求前缀和的时间复杂读也是logn,因此单次操作的时间复杂度为(logn)^2
代码:
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e5;
int tree[N + 5],pre[N + 5],high[N + 5]; //分别存储树状数组、输入的数组、结果数组
int n;
void update(int x,int y){ //更新树状数组
for(;x <= n;x += x & -x)
tree[x] += y;
}
int get_sum(int x){ //前缀和
int ans = 0;
for(;x;x -= x & -x)
ans += tree[x];
return ans;
}
signed main(){
cin>>n;
for(int i = 2;i <= n;++i)
cin>>pre[i];
for(int i = 1;i <= n;++i)
update(i,1);
for(int i = n;i >= 1;--i){
int l = 1,r = n,mid;
while(l < r){ //找到最左边的相等的点
mid = (l + r) >> 1;
if(get_sum(mid) < 1 + pre[i])
l = mid + 1;
else if(get_sum(mid) > 1 + pre[i])
r = mid - 1;
else
r = mid;
}
high[i] = l;
update(l,-1); //去除该身高
}
for(int i = 1;i <= n;++i)
cout<<high[i]<<"\n";
return 0;
}