poj-2828

最新推荐文章于 2019-08-14 10:13:38 发布

fyfcauc

最新推荐文章于 2019-08-14 10:13:38 发布

阅读量459

点赞数

分类专栏： POJ 线段树文章标签： poj 线段树

本文链接：https://blog.csdn.net/fyfcauc/article/details/38317407

版权

POJ 同时被 2 个专栏收录

233 篇文章 0 订阅

订阅专栏

线段树

13 篇文章 0 订阅

订阅专栏

// 14460K   3766MS  G++
#include <stdio.h>
#include <string.h>

#define MAX 200100

struct TreeNode {
    int val;
    int l;
    int r;
    int insertPosNum;
};

typedef struct TreeNode TreeNode;

TreeNode tree[MAX*4];

int peopleNum;

struct PeopleJumpInfo {
    int pos;
    int val;
};

typedef struct PeopleJumpInfo PeopleJumpInfo;
PeopleJumpInfo jump[MAX];

int curFreeNodeId;

void buildTree(int l, int r, int pos) {
    tree[pos].l = l;
    tree[pos].r = r;
    tree[pos].val = -1;
    tree[pos].insertPosNum = r - l + 1;
    if (l == r) {
        return;
    } else {
        int mid = (l+r)>>1;
        buildTree(l, mid, pos<<1);
        buildTree(mid+1, r, pos<<1|1);
    }
}

void insert(int pos, int val, int nodeId) {
    // printf("insert %d %d\n", tree[nodeId].r, tree[nodeId].l);
    if (tree[nodeId].r == tree[nodeId].l) { // has narrow to 1 pos.
        tree[nodeId].val = val;
        tree[nodeId].insertPosNum = 0;
    } else {
        int leftInsertPosNum = tree[nodeId<<1].insertPosNum;
        int rightInsetyPosNum = tree[nodeId<<1|1].insertPosNum;
        tree[nodeId].insertPosNum--;
        if (leftInsertPosNum >= pos) {
            insert(pos, val, nodeId<<1);
        } else if (leftInsertPosNum < pos) {
            insert(pos - leftInsertPosNum, val, nodeId<<1|1);
        }
    }
}

int nodeNum;

void printFinalQueue(int nodeId) {
    if (tree[nodeId].l == tree[nodeId].r) {
        if (tree[nodeId].val >= 0) {
            nodeNum++;
            if (nodeNum < peopleNum) {
                printf("%d ", tree[nodeId].val);
            } else if (nodeNum == peopleNum){
                printf("%d\n", tree[nodeId].val);
            }
        }
        return;
    } else {
        printFinalQueue(nodeId<<1);
        printFinalQueue(nodeId<<1|1);
    }
}

int main() {
    while(scanf("%d", &peopleNum) != EOF) {
        memset(tree, 0, sizeof(tree));
        // for (int i = 0; i < MAX; i++) {
        //     tree[i].val = -1;
        // }
        nodeNum = 0;
        curFreeNodeId = 0;
        for (int i = 0; i < peopleNum; i++) {
            int pos, val;
            scanf("%d %d", &pos, &val);
            jump[i].pos = pos + 1;
            jump[i].val = val;
        }
        buildTree(1, peopleNum, 1); // root range: 1 <-> peoplenum
        // printf("buildTree\n");
        for (int i = peopleNum - 1; i >=0; i--) {
            insert(jump[i].pos, jump[i].val, 1);
        }
        // printf("printFinalQueue\n");
        printFinalQueue(1);
        // printf("\n");
    }
}

从纯线段树角度看，算是水题，但是从转化上看，感觉难度还是挺高的，唉，可能是自己悟性差，不过无所谓了，反正我也没跟别人比的资格。

最朴素的思路是模拟大法，但是绝壁TLE，因此直接pass，后来想过用二叉树，这样二叉树的每个节点代表一个人，root是第一次进入队列的那个人，同时二叉树动态维护自己当前在队列的名次，这样以后有新的插队者，就不用像纯模拟那样的挪动整个数组了，只需插到合适的二叉树位置即可（左child是前一个位置，右child是后一个位置），最后做一个中序遍历就可以了，不过后来发现可能维护这个名次会比较费劲，对某个树左边插入一个新节点，该树的整个右子树都要更新，估计最后也是TLE的命。

最开始怎么都想不到用线段树，后来也是搜了下，才大概清楚，不过也用到了一些技法：逆序插入，同时线段树的每个节点，保存的是该节点对应的区间[l, r]还有多少个空缺可以填（一开始每个区间必然都有 r - l + 1个空位可填），然后逆序的将人插入到线段树中，一直向下插，直到遇到点（就是长度为1的区间，线段点），插入的操作是这样的：

比如有4个人：

原始数据：

4
0 77
1 51
1 33
2 69

首先简单转化一下，从插入的前一个位置变为插入的位置，+1即可：

1 77

2 51

2 33

3 69

那么逆序就是

3 69

3 33

3 51