PAT A1051 Pop Sequence (25 分)

题意

给定栈空间和入栈顺序，对出栈顺序的合法性进行判断

题目注意点

• 每次读取一行数据时都要清空栈
• 另外用一个数组存储出栈序列，防止判错时让后面的数字错位
• 入栈顺序用第二层循环表示，每次压栈一次之后判断如何出栈，而不是读取了出栈序号在思考如何压栈

题解

算法笔记：

#include <stdio.h>
#include <stack>
using namespace std;
const int maxn = 1010;
int a[maxn];
stack<int> st;
int main() {
    int m, n, T;
    scanf("%d%d%d", &m, &n, &T);
    while (T--) {
        while (!st.empty()) {//清空栈
            st.pop();
        }
        for (int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
        }
        bool flag = true;
        int current = 1;
        for (int i = 1; i <= n; i++) {
            st.push(i);
            if (st.size() > m) {
                flag = false;
                break;
            }
            while (!st.empty() && st.top() == a[current]) {
                st.pop();
                current++;
            }
        }
        if (st.empty() && flag) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}