Inna and Sequence CodeForces - 374D

最新推荐文章于 2019-10-22 16:56:42 发布

sunyutian1998

最新推荐文章于 2019-10-22 16:56:42 发布

阅读量245

点赞数

CC 4.0 BY-SA版权

分类专栏：线段树/树状数组/RMQ 文章标签：线段树/树状数组/RMQ CodeForces

本文链接：https://blog.youkuaiyun.com/sunyutian1998/article/details/83420009

线段树/树状数组/RMQ 专栏收录该内容

124 篇文章

订阅专栏

本文提供了一种使用线段树解决 Codeforces 374D 问题的方法，通过预处理和二分查找来高效地处理删除操作，确保了算法的效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://codeforces.com/problemset/problem/374/D

预处理每个-1操作要删哪几个数并存下来这些操作数肯定是<=n的然后线段树维护整个序列然后扫一遍所有-1操作每次二分查找来删掉对应位置的数即可

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
const int maxm=1e6+10;

struct node
{
    int val,gou;
};

node tree[4*maxn];
int pos[maxm],ary[maxn],pre[maxn];
int n,m;

void pushup(int cur)
{
    tree[cur].val=tree[2*cur].val+tree[2*cur+1].val;
}

void build(int l,int r,int cur)
{
    int m;
    if(l==r){
        if(ary[l]==-1) tree[cur].val=0,tree[cur].gou=-1;
        else tree[cur].val=1,tree[cur].gou=ary[l];
        return;
    }
    m=(l+r)/2;
    build(l,m,2*cur);
    build(m+1,r,2*cur+1);
    pushup(cur);
}

void update(int tar,int val,int l,int r,int cur)
{
    int m;
    if(l==r){
        if(val==-1) tree[cur].val=0,tree[cur].gou=-1;
        else tree[cur].val=1,tree[cur].gou=val;
        return;
    }
    m=(l+r)/2;
    if(tar<=tree[2*cur].val) update(tar,val,l,m,2*cur);
    else update(tar-tree[2*cur].val,val,m+1,r,2*cur+1);
    pushup(cur);
}

void safari(int l,int r,int cur)
{
    int m;
    if(l==r){
        if(tree[cur].val) printf("%d",tree[cur].gou);
        return;
    }
    m=(l+r)/2;
    safari(l,m,2*cur);
    safari(m+1,r,2*cur+1);
}

int main()
{
    int i,j,cnt,p;
    scanf("%d%d",&n,&m);
    for(i=1;i<=m;i++) scanf("%d",&pos[i]);
    for(i=1;i<=n;i++) scanf("%d",&ary[i]);
    cnt=0;
    for(i=1;i<=n;i++){
        if(ary[i]==-1){
            p=upper_bound(pos+1,pos+m+1,cnt)-pos-1;
            pre[i]=p,cnt-=p;
        }
        else cnt++;
    }
    /*
    for(i=1;i<=n;i++){
        if(ary[i]==-1) printf("%d ",pre[i]);
    }
    printf("\n");
    */
    if(cnt!=0)
    {
        build(1,n,1);
        for(i=1;i<=n;i++){
            if(ary[i]==-1){
                for(j=1;j<=pre[i];j++){
                    update(pos[j]-j+1,-1,1,n,1);
                }
            }
        }
        safari(1,n,1);
    }
    else printf("Poor stack!\n");
    return 0;
}