(线段树)POJ2828 Buy Tickets

一道并不是太难的线段树练习题。

题意大概是将一些数字插入一个序列中,每当插到一个位置之后,会把当前以及后面的数字都往后移一格,最后输出他们的顺序。由于每次插入都会影响到其他数字的位置,因此越靠后的插入,自身的位置越不容易被影响——而最后一个数插到哪,最终位置就是哪。因此最好的办法就是从后往前推。
假设一共有n个人,对于倒数第i个人而言,如果他要插在第k个位置上,那么只需要从前往后数到第k个空位,然后把它插入就可以了;后面已经处理过的i-1个人的插入对于他而言没有影响。线段树的random域就可以用来保存该段区间的空位数。所以问题变成,找到第 i 个空位,线段树记录该区间里有几个空位。

     具体代码如下:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
typedef long long ll;

const int maxn = 200010;

int n;

int sum[4*maxn+10];//以数组形式建立线段树,存储空位数
int pos[maxn+1], val[maxn+1];
int ans[maxn+1];//存储答案

void pushUp(int root) {
    sum[root] = sum[root<<1] + sum[root<<1|1];
    return;
}

void build(int root, int l, int r) {//建树
    if(l == r) {
        sum[root] = 1;
        return;
    }
    else {
        int m = (l+r) >> 1;
        build(root<<1, l, m);
        build(root<<1|1, m+1, r);
        pushUp(root);
        return;
    }
}

int query(int root, int l, int r, int pos) {
    if(l == r) {//找到空位,把这个位占用
        sum[root] = 0;
        return l;
    }
    else {
        int m = (l+r) >> 1;
        int rank;//空位
        if(sum[root<<1] > pos)//在左子树中找空位
            rank = query(root<<1, l, m, pos);
        else
            rank = query(root<<1|1, m+1, r, pos-sum[root<<1]);
        pushUp(root);
        return rank;
    }
}

int main() {
    while(~scanf("%d", &n)) {
        for(int i = 1;i <= n;i++)
            scanf("%d %d", &pos[i], &val[i]);
        build(1, 0, n-1);
        for(int i = n;i > 0;i--) {
            int res = query(1, 0, n-1, pos[i]);//查找插入的空位
            ans[res] = val[i];//将相应权值存入所查到的下标对应的ans数组中
        }
        for(int i = 0;i < n-1;i++)
            printf("%d ", ans[i]);
        printf("%d\n", ans[n-1]);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值