判断栈的输出序列是否合法

问题简述

一个最多可以存储M个数的栈；按1,2,3...顺序入栈并随机出栈
输入一个出栈序列，判断给出 出栈序列是否合理

大概需要用的元素

一个栈——按序入栈
一个数——判断栈的大小是否超出要求
一个flag——标志该序列是否合理

代码

#include<stdio.h>
#include<stack>

using namespace std;
const int maxn = 1010;
int sqe[maxn];

int main()
{
    int N, M, K;
    stack<int> st;
    scanf("%d%d%d", &N, &M, &K);
    while (K--)
    {
        while(!st.empty())
        {
            st.pop();
        }
        for(int i = 1; i <= N; i++)
        {
            scanf("%d", &sqe[i]);
        }
        int cur_idx = 1;
        bool flag = true;           // 合法标志
        for (int i = 1; i <= N; i++)
        {
            st.push(i);
            if(st.size() > M)
            {
                flag = false;
                break;              // 不合法，则跳出本次循环
            }
            if(!st.empty() && st.top() == sqe[cur_idx]) // 注意，top() 和 pop() 都需要判断是否为空
            {
                st.pop();
                cur_idx++;
            }
        }
        if(st.empty() && flag == true)
        {
            printf("Yes\n");
        }
        else
        {
            printf("No\n");
        }
    }

    getchar();
    getchar();
    return 0;
}