poj 2828 插队运用(线段树,单点更新)

题意:

给n,以下n行给出一个位置p,一个权值v。

代表了这个人v插队在位置p。

最后输出队伍。


解析:

归类在线段树里面,怎么也想不通用线段树来解啊。

这题网上主流的解法是将整个输入倒回来往空队列里面插。

因为倒回来,最后进去的人v肯定在的位置是p,所以直接插进去。

线段树中维护一个空位置的数组,代表着一段间还有多少个空位置。

插入的时候,检查,若左子树的空位置大于当前插入的位置,则向左子树递归找到插入点就好了;

若左子树的空间已经不足以装下这个点的位置,则向右子树寻找插入点,但插入的位置需要减掉前一段子树开始计位的点。


代码:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#include <vector>
#include <queue>
#include <map>
#include <climits>
#include <cassert>
#define LL long long
#define lson lo, mi, rt << 1
#define rson mi + 1, hi, rt << 1 | 1

using namespace std;
const int maxn = 200000 + 10;
const int inf = 0x3f3f3f3f;
const double eps = 1e-8;
const double pi = acos(-1.0);
const double ee = exp(1.0);

int spare[maxn << 2];
int que[maxn];
int pos[maxn], val[maxn];

void pushup(int rt)
{
    spare[rt] = spare[rt << 1] + spare[rt << 1 | 1];
}

void build(int lo, int hi, int rt)
{
    if (lo == hi)
    {
        spare[rt] = 1;
        return;
    }
    int mi = (lo + hi) >> 1;
    build(lson);
    build(rson);
    pushup(rt);
}

void update(int p, int v, int lo, int hi, int rt)
{
    if (lo == hi)
    {
        que[lo] = v;
        spare[rt] = 0;
        return;
    }
    int mi = (lo + hi) >> 1;
    if (p <= spare[rt << 1])
        update(p, v, lson);
    else
        update(p - spare[rt << 1], v, rson);
    pushup(rt);
}

int main()
{
    #ifdef LOCAL
    freopen("in.txt", "r", stdin);
    #endif // LOCAL
    int n;
    while (scanf("%d", &n) != EOF)
    {
        build(1, n, 1);
        for (int i = 1; i <= n; i++)
        {
            scanf("%d %d", &pos[i], &val[i]);
        }
        for (int i = n; i > 0; i--)
        {
            update(pos[i] + 1, val[i], 1, n, 1);
        }
        for (int i = 1; i <= n; i++)
        {
            printf("%d ", que[i]);
        }
        printf("\n");
    }
    return 0;
}


你可能感兴趣的:(poj 2828 插队运用(线段树,单点更新))