Codeforces 374D - Inna and Sequence

374D - Inna and Sequence

思路：

树状数组+二分

因为被删的点最多N=1e6个，所以复杂度N*logN*logN

前段时间做过一道一样的题，这类题基本套路二分找没删除前的位置

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int N=1e6+;
int n;
int bit[N];
int a[N];
int ans[N];
bool vis[N];
vector<int>v;
void add(int x,int a){
    while(x<=1e6)bit[x]+=a,x+=x&-x;
}
int sum(int x){
    int ans=;
    while(x)ans+=bit[x],x-=x&-x;
    return ans;
}
int srch(int t){
    int l=,r=1e6,mid=(l+r)>>;
    while(l<r){
        if(sum(mid)>=t)r=mid;
        else l=mid+;
        mid=(l+r)>>;
    }
    return mid;
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie();
    for(int i=;i<=1e6;i++)add(i,);
    int cnt=,t,m;
    cin>>n>>m;
    for(int i=;i<m;i++)cin>>a[i];
    for(int i=;i<n;i++){
        cin>>t;
        if(t==-){
            v.clear();
            for(int j=;j<m;j++){
                int l=srch(a[j]);
                if(l>cnt)break;
                v.pb(l);
            }
            for(int j=;j<v.size();j++){
                vis[v[j]]=true;
                add(v[j],-);
            }
        }
        else{
            ans[++cnt]=t;
        }
    }
    if(sum(cnt)==)cout<<"Poor stack!"<<endl;
    else{
        for(int i=;i<=cnt;i++)if(!vis[i])cout<<ans[i];
        cout<<endl;
    }
    return ;
}