「SuperOJ 142」车厢调度

5
3 1 2 4 5

NO

#include <cstdio>
#include <iostream>
#include <stack>
using namespace std;
stack<int> st;
inline bool test(int* arr, int len) {
    int pos = 0;
    for (int i = 1; i <= len; i++) {
        st.push(i);
        while (!st.empty()) {
            if (st.top() != arr[pos])
                break;
            else {
                st.pop();
                pos++;
            }
        }
    }
    return st.empty();
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    int n;
    cin >> n;
    int* arr = new int[n];
    for (int i = 0; i < n; i++) cin >> arr[i];
    if (test(arr, n))
        cout << "YES\n";
    else
        cout << "NO\n";
    return 0;
}