线段树(单点更新)poj2828

来源:互联网 发布:淘宝新店开直通车 编辑:程序博客网 时间:2024/05/22 05:00

题意:

输入n个有序对<pi, vi> pi表示在第pi个位置后面插入一个值为vi的人,并且pi是不降的。

输出最终得到的v的序列。

线段树,记录区间内的空位置数,反过来将人插入到序列中。

#include <iostream>#include <cstdio>using namespace std;#define N 200005//spare存储区间内的空位置数int spare[N << 2], seq[N];void pushUp(int rt){    spare[rt] = spare[rt << 1] + spare[rt << 1 | 1];}void build(int l, int r, int rt){    if (l == r) {        spare[rt] = 1; return; //初始时,子节点的空位置数均为1    }    int m = (l + r) >> 1;    build(l, m, rt << 1);    build(m + 1, r, rt << 1 | 1);    pushUp(rt);}void update(int p, int v, int l, int r, int rt){    if (l == r) {        seq[l] = v; spare[rt] = 0; return; //插入元素后将该子节点的空位置为0    }    /*      若左子树的空格数大于等于要放入的位置,则搜索左子树;      否则,搜索右子树,由于左边有一些空位置,只需要搜索      右子树的部分空位置,使左右空位置之和为要搜索的位置。    */    int m = (l + r) >> 1;    if (spare[rt << 1] >= p) update(p, v, l, m, rt << 1);    else update(p - spare[rt << 1], v, m + 1, r, rt << 1 | 1);    pushUp(rt); //向上更新}int main(){    int n, p[N], v[N], i;    while (scanf ("%d", &n) != EOF)    {        build (1, n, 1);        for (i = 1; i <= n; i++)             scanf ("%d %d", &p[i], &v[i]);        /*        倒过来插入,因为这样插入可以轻易确定所需插入元素的最终位置。        设目前插入元素为x,插入到第k个元素之后,插入的元素的最终位置        就是第k+1个空位置处。        */        for (i = n; i > 0; i--)             update(p[i] + 1, v[i], 1, n, 1);        for (i = 1; i <= n; i++)             printf ("%d ", seq[i]);        printf ("\n");    }    return 0;}


原创粉丝点击