PAT|1051 Pop Sequence

题目大意

给定一个堆栈，最多可以保留M个数字。按1、2、3，…，N的顺序按N个数字，然后随机弹出。您应该判断给定的数字序列是否是堆栈的可能弹出序列。例如，如果M为5，N为7，我们可以从堆栈中获得1、2、3、4、5、6、7，但不能获得3、2、1、7、5、6、4。

柳神代码

#include <iostream>
#include <stack>
#include <vector>
using namespace std;
int main() {
    int m, n, k;
    scanf("%d %d %d", &m, &n, &k);
    for(int i = 0; i < k; i++) {
        bool flag = false;
        stack<int> s;
        vector<int> v(n + 1);
        for(int j = 1; j <= n; j++)
            scanf("%d", &v[j]);
        int current = 1;
        for(int j = 1; j <= n; j++) {
            s.push(j);
            if(s.size() > m) break;
            while(!s.empty() && s.top() == v[current]) {
                s.pop();
                current++;
            }
        }
        if(current == n + 1) flag = true;
        if(flag) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}