码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 2182 Lost Cows

时间:2018-03-20 21:46:32      阅读:211      评论:0      收藏:0      [点我收藏+]

标签:down   倍增   数组   query   post   out   cout   clu   span   

树状数组 + 倍增

动态维护第k大
注意倍增是 < x ,不能等于防止重复
本题从后往前扫,维护一个 01 序列,对于已经确定的排名,将他变成 0 即变成了维护第 k 大的数
用倍增结合了树状数组相当好的性质

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
const int MAXN = 100005;
int n, c[MAXN], lg[MAXN], po[MAXN], sum, ans, num[MAXN], t[MAXN];
int lowbit(const int &x) {
    return x & -x;
}
void add(int x, int y) {
    for( ; x <= n; x += lowbit(x)) c[x] += y;
}
int query(int x) {
    sum = ans = 0;
    for(int i = lg[n]; i >= 0; i--) {
        if(ans + po[i] <= n && sum + c[ans + po[i]] < x) ans += po[i], sum += c[ans];
    }
    return ans + 1;
}
int main() {
    cin >> n;
    lg[2] = 1;
    for(int i = 3;i <= 2 * n; i++) lg[i] = lg[i >> 1] + 1;
    po[0] = 1;
    for(int i = 1; i <= 16; i++) po[i] = 1 << i;
    t[1] = 0;
    for(int i = 2; i <= n; i++) {
        cin >> t[i];
    } 
    for(int i = 1; i <= n; i++) add(i, 1);
    for(int i = n; i >= 1; i--) {
        num[i] = query(t[i] + 1);
        add(num[i], -1);
    }
    for(int i = 1; i <= n; i++) cout << num[i] << endl;
    return 0;
}

POJ 2182 Lost Cows

标签:down   倍增   数组   query   post   out   cout   clu   span   

原文地址:https://www.cnblogs.com/Mr-WolframsMgcBox/p/8612530.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!