Lost Cows POJ - 2182 二分 + 树状数组

Code:

#include<cstdio>
#include<stack>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 100000 + 233;
int n, C[maxn], height[maxn];
stack<int>S;
inline int lowbit(int t) { return t & (-t);}
inline void update(int x, int delta)
{
    while(x <= n) C[x] += delta, x += lowbit(x);
}
inline int query(int x)
{
    int sum = 0;
    while(x > 0) sum += C[x], x -=lowbit(x);
    return sum;
}
int main()
{
    scanf("%d",&n);
    height[1] = 0;
    for(int i = 2;i <= n; ++i) scanf("%d",&height[i]);
    for(int i = 1;i <= n; ++i) update(i, 1);
    for(int i = n;i >= 1; --i)
    {
        int l = 1, r = n, ans = 0;
        while(l <= r)
        {
            int mid = (l + r) >> 1;
            if(query(mid) >= height[i] + 1) ans = mid, r = mid - 1;
            else l = mid + 1;
        }
        update(ans, -1);
        S.push(ans);
    }
    while(!S.empty()) {printf("%d\n",S.top()); S.pop();}
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值