Codeforces760 E. Nikita and stack（分块）

Posted 2021-07-23 live4m

题意：

解法：

分块.

每次操作暴力修改一个块,维护多pop的次数pp[x],和栈stk[x].

逆序遍历每个块,遍历的时候维护右边遗留的pop次数,如果当前栈没有被全部pop掉则可以输出答案.

算法总复杂度O(n*sq).

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=1e6+5;
int op[maxm],v[maxm];
int l[maxm],r[maxm];
int belong[maxm];
vector<int>s[maxm];
int pp[maxm];
int block,num;
int n;
void init(){
    block=sqrt(n);
    num=n/block;
    if(n%block)num++;
    for(int i=1;i<=num;i++){
        l[i]=r[i-1]+1;r[i]=l[i]+block-1;
    }
    r[num]=n;
    for(int i=1;i<=num;i++){
        for(int j=l[i];j<=r[i];j++){
            belong[j]=i;op[j]=-1;
        }
    }
}
void rebuild(int x){
    s[x].clear();
    pp[x]=0;
    for(int i=l[x];i<=r[x];i++){
        if(op[i]==0){
            if(s[x].size())s[x].pop_back();
            else pp[x]++;
        }else if(op[i]==1){
            s[x].push_back(v[i]);
        }
    }
}
void update(int p,int t){
    op[p]=t;
    if(t==1)cin>>v[p];
    rebuild(belong[p]);
}
int ask(){
    int lastp=0;
    for(int i=num;i>=1;i--){
        int len=s[i].size();
        if(lastp)len-=lastp;
        if(len-1>=0){
            return s[i][len-1];
        }else{
            lastp-=s[i].size();
            lastp+=pp[i];
        }
    }
    return -1;
}
void solve(){
    cin>>n;
    init();
    for(int i=1;i<=n;i++){
        int p,t;cin>>p>>t;
        update(p,t);
        int ans=ask();
        cout<<ans<<endl;
    }
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    solve();
    return 0;
}

---