谜一样的牛【树状数组+二分】

谜一样的牛【树状数组+二分】

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;
}