poj 2828 Buy Tickets

最新推荐文章于 2021-04-10 19:11:55 发布

原创最新推荐文章于 2021-04-10 19:11:55 发布 · 244 阅读

1 ·

CC 4.0 BY-SA版权

线段树专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用线段树的数据结构解决特定排队问题的方法。通过逆序输入数据并利用线段树维护区间内空位数量，实现高效查找与更新操作。适用于需要快速查询和更新区间元素的应用场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://poj.org/problem?id=2828

思路：将输入数据逆序传入，pos[i]表示此时这个人前面应有多少个空位。所以就用线段树来维护某个区间内的空位个数。在查找操作时如果t[rt<<1]<=target，就将target减去t[rt<<1]，在右子树中找，最终定位在target==0且l==r的位置。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <queue>
#define fi first
#define se second
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
using namespace std;
typedef long long LL;
typedef pair<int, int> P;
//head
const int MAX = 2e5+5;
P q[MAX];
int t[MAX<<2];
int ans[MAX];

void pushUp(int rt)
{
    t[rt] = t[rt<<1] + t[rt<<1|1];
}

void build(int l, int r, int rt)
{
    if(l==r)
    {
        t[rt] = 1;
        return;
    }
    int m = (l+r)>>1;
    build(lson);
    build(rson);
    pushUp(rt);
}

void update(int pos, int l, int r, int rt)
{
    if(l==r)
    {
        t[rt]--;
        return;
    }
    int m = (l+r)>>1;
    if(pos<=m)
        update(pos, lson);
    else
        update(pos, rson);
    pushUp(rt);
}

int query(int target, int l, int r, int rt)//return the position to add the entry
{
    if(l==r && target==0)
        return r;
    int m = (l+r)>>1;
    if(t[rt<<1]<=target)
        return query(target-t[rt<<1], rson);
    else
        return query(target, lson);
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        build(1,n,1);
        for(int i=1; i<=n; i++)
            scanf("%d%d",&q[i].fi,&q[i].se);
        for(int i=n; i>=1; i--)
        {
            int pos = query(q[i].fi, 1, n, 1);
            ans[pos] = q[i].se;
            update(pos, 1, n, 1);
        }
        for(int i=1; i<n; i++)
            printf("%d ",ans[i]);
        printf("%d\n",ans[n]);
    }
    return 0;
}