数据结构——火车站中驶出的火车

火车站中驶出的火车

描述
有编号从1到N的N辆火车等待进入车站，而车站的容量是有限的，需要对火车进行调度。只有两种调度指令，进站和出站。现在车站给出了M个指令，请你输出在完成这些调度指令后，车站中驶出火车编号。

输入
两行
第一行两个整数，N和M
第二行为M个为-1或者为1的整数，-1代表出站，1代表入站

输出
一行整数，为驶出车站的火车编号

样例输入
3 3
1 -1 1

样例输出
1

#include<iostream>
using namespace std;
int a[10001];
int main(){
    int m,n;
    int count=0;
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        int j;
        cin>>j;
        if(j==1){
            count++;
        }
        else if(j==-1){
            cout<<count<<" ";
            count--;
        }
    }
}