POJ2828-Buy Ticket(权值线段树)

题目链接:POJ2828-Buy Tickets

题意:

有n个人依次插队,每次给出两个值u,v,表示代号为v的人插在了第u个人的后面,询问最终序列。

题解:

因为最开始的序列一定是空的,所以每当我们插入一个新的v时,他前面一定有u个人,所以我们可以倒着考虑:每个人的位置就是固定的了,每当插入一个新的v时,前面一定有u个空位置。

线段树维护即可

参考代码:

#include <iostream>
#include <cstdio>
using namespace std;
typedef long long ll;
const int MAXN = 2e5 + 5;
int n;
int u[MAXN], v[MAXN], a[MAXN];
int sum[MAXN << 2];
void pushup(int rt)
{
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
}
void build(int l, int r, int rt)
{
    if (l == r)
    {
        sum[rt] = 0;
        return;
    }
    int mid = (l + r) / 2;
    build(l, mid, rt << 1);
    build(mid + 1, r, rt << 1 | 1);
    pushup(rt);
}
void update(int p, int val, int l, int r, int rt)
{
    if (l == r)
    {
        sum[rt] = 1;
        a[l] = val;
        return;
    }
    int mid = (l + r) >> 1;
    int temp = sum[rt << 1];
    temp = mid - l + 1 - temp;
    if (p <= temp)
        update(p, val, l, mid, rt << 1);
    else
        update(p - temp, val, mid + 1, r, rt << 1 | 1);
    pushup(rt);
}
int main()
{
    while (scanf("%d", &n) == 1)
    {
        build(1, n, 1);
        for (int i = 1; i <= n; i++)
            scanf("%d%d", &u[i], &v[i]);
        for (int i = n; i >= 1; i--)
            update(u[i] + 1, v[i], 1, n, 1);

        for (int i = 1; i <= n; i++)
            printf("%d ", a[i]);
        puts("");
    }
    return 0;
}

发表留言

人生在世,错别字在所难免,无需纠正。