PAT A1051 Pop Sequence (25 分) 栈

    题目大意：给出一个容量为M的栈，将数字1~N依次执行Push和Pop操作，判断给出的序列是否可能是Pop序列。

    直接用STL中的栈进行模拟，对给出的序列进行遍历，如果当前数字大于栈顶元素或栈空，则执行压栈操作，从上次被压入的最后一个数+1开始，一直到当前数字，然后pop，持续判断栈的大小是否超过M。如果当前数字小于栈顶元素，则不可能。当整个序列遍历完后，stack应为空，否则也是不可能的序列。

AC代码：

#include <iostream>
#include <vector>
#include <cstdio>
#include <stack>

using namespace std;

bool judge(vector<int> &v, int M)
{
    stack<int> st;
    int lastPush = 0;
    for (int i = 0; i < v.size(); ++i)
    {
        if(st.empty() || st.top() <= v[i])
        {
            for (int j = lastPush + 1; j <= v[i]; ++j)
            {
                st.push(j);
                lastPush = j;
                if(st.size() > M) return false;
            }
        }
        else return false;
        st.pop();
    }
    if(!st.empty()) return false;
    return true;
}

int main()
{
    int M, N, K;
    cin >> M >> N >> K;
    for (int query = 0; query < K; ++query)
    {
        vector<int> v(N);
        for (int i = 0; i < N; ++i)
            scanf("%d", &v[i]);
        if(judge(v, M)) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}