C++一维数组（博物馆）

Question

这道题目用C++怎么写
我的错误代码

#include<cstdio>
long long n,a[101992],x,sum=0,m;
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&x);
        if(x==0){
            scanf("%d",&a[i]);
            sum=sum+a[i]+1; 
        }
        if(x==1){
            int y=0;
            scanf("%d",&y);//将y当成第几批人离开 
            sum=sum-a[y]-1;//第y批人离开-1 
    }
    printf("%d",sum);
}

此代码无法实现以下操作：

如何解决

Answer 1

#include<iostream>
using namespace std;
int a[10000001],n,x,q1,q2,s;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>x;
        if(!x){
            q1++;
            cin>>a[q1];
            a[q1]++;
        }else{
            cin>>q2;
            a[q2]=0;
        }
    }for(int i=1;i<=n;i++)s+=a[i];
    cout<<s;
    return 0;
}