CodeForces - 374D Inna and Sequence 线段树

原创于 2019-10-22 16:56:42 发布 · 194 阅读

0 ·

CC 4.0 BY-SA版权

线段树专栏收录该内容

68 篇文章

订阅专栏

本文解析了CodeForces-374D题目的算法解决方案，通过使用线段树数据结构来高效处理一系列操作，包括插入和删除元素，确保复杂度保持在nlogn级别。文章详细介绍了如何构建和更新线段树，以及如何实现删除操作的判断逻辑。

题目连接：https://vjudge.net/problem/CodeForces-374D

题意：n次操作，先给出m个数，原先有一个空序列，每次操作给出一个数，为1或0都放在后面，为-1时，删除m个位置的数

题解：只看n和m是很大的不能直接操作，但是单看每个数的贡献来说，对于每个数，插入和删除只有两次，所以复杂度是nlogn，那就看怎么删除了，删除的时候就看剩下的数够不够删的就可了

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
struct node {
    int l, r;
    int op;
    int num, laz;
}tree[N << 2];
int n, m, a[N];
void build(int l, int r, int cur) {
    tree[cur].l = l;
    tree[cur].r = r;
    tree[cur].op = -1;
    tree[cur].laz = r - l + 1;
    if(l == r) return;
    int mid = (l + r) >> 1;
    build(l, mid, cur << 1);
    build(mid + 1, r, cur << 1 | 1);
}
int flag;
void pushup(int cur) {
    tree[cur].laz = tree[cur << 1].laz + tree[cur << 1 | 1].laz;
    tree[cur].num = tree[cur << 1].num + tree[cur << 1 | 1].num;
}
void update1(int k, int cur, int op) {
    if(tree[cur].l == tree[cur].r) {
        tree[cur].laz = 0;
        tree[cur].num = 0;
        tree[cur].op = op;
        return;
    }
    if(k <= tree[cur << 1].num) update1(k, cur << 1, op);
    else update1(k - tree[cur << 1].num, cur << 1 | 1, op);
    pushup(cur);
}
void update2(int pos, int cur, int op) {
    if(tree[cur].l == tree[cur].r) {
        tree[cur].laz = 0;
        tree[cur].num = 1;
        tree[cur].op = op;
        return;
    }
    if(pos <= tree[cur << 1].r) update2(pos, cur << 1, op);
    else update2(pos, cur << 1 | 1, op);
    pushup(cur);
}
void print(int cur) {
    if(tree[cur].l == tree[cur].r) {
        printf("%d", tree[cur].op);
        return;
    }
    if(tree[cur << 1].num) print(cur << 1);
    if(tree[cur << 1 | 1].num) print(cur << 1 | 1);
}
int main() {
    int pre = 0;
    int x;
    scanf("%d %d", &n, &m);
    for(int i = 1; i <= m; i++) scanf("%d", &a[i]);
    build(1, n, 1);
    for(int i = 1; i <= n; i++) {
        scanf("%d", &x);
        if(x == -1) {
            if(tree[1].num == 0) continue;
            for(int j = 1; j <= m; j++) {
                if(tree[1].num < a[j] - j + 1) break;
                update1(a[j] - j + 1, 1, -1);
            }

        } else {
            update2(pre + 1, 1, x);
            pre++;
        }
    }
    if(tree[1].num == 0) {
        printf("Poor stack!\n");
        return 0;
    }
    print(1);
    return 0;
}