C. Nikita and stack(栈操作 想法 线段树)

我与影子孤独终老i 提交于 2019-12-02 08:40:26

http://codeforces.com/problemset/problem/756/C

题意: 给出n个操作,每个有一个序号。对于每个前缀,求出这些操作按照序号排序后进行操作,结果中栈顶元素是什么。

解析: 考虑序号为x的操作,压入元素y,对于包含x的所有前缀,序号大于x的操作刚好完美匹配时,栈顶元素才为y。

  • 序号为x的压入操作,区间[1,x]加1。
  • 序号为x的弹出操作,区间[1,x]减1。

每次操作完后最右边的正数所在的位置就是答案的序号。

理由:

  • 序号大于x的操作,若存在一个时刻y,[x+1,y]弹出的次数大于[x+1,y]压入的次数,则x压入的数被弹出,表现为a[x]<=0
  • 序号大于x的操作,若[x+1,inf)压入大于弹出,则说明x上面还有元素。表现为:x存在其它正数。

代码:

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int mx[N*4],flag[N*4],a[N];
void push_down(int root)
{
    if(!flag[root])return ;
    mx[root<<1]+=flag[root];
    mx[root<<1|1]+=flag[root];
    flag[root<<1]+=flag[root];
    flag[root<<1|1]+=flag[root];
    flag[root]=0;
}
void update(int l,int r,int root,int ql,int qr,int v)
{
    if(l>=ql&&r<=qr)
    {
        mx[root]+=v;
        flag[root]+=v;
        return ;
    }
    int mid=l+r>>1;
    push_down(root);
    if(mid>=ql)
        update(l,mid,root<<1,ql,qr,v);
    if(mid<qr)
        update(mid+1,r,root<<1|1,ql,qr,v);
    mx[root]=max(mx[root<<1],mx[root<<1|1]);
}
int query(int l,int r,int root)
{
    if(mx[1]<=0)return -1;
    if(l==r)return a[l];
    int mid=l+r>>1;
    push_down(root);
    if(mx[root<<1|1]>=1)return query(mid+1,r,root<<1|1);
    else return query(l,mid,root<<1);
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int op,x,y;
        scanf("%d%d",&x,&op);
        if(op-1)
            update(1,n,1,1,x,-1);
        else
            scanf("%d",&y),update(1,n,1,1,x,1),a[x]=y;
        printf("%d\n",query(1,n,1));
    }
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!