PTA（出栈序列的合法性）

题目：

给定一个最大容量为 M 的堆栈，将 N 个数字按 1, 2, 3, ..., N 的顺序入栈，允许按任何顺序出栈，则哪些数字序列是不可能得到的？例如给定 M=5、N=7，则我们有可能得到{ 1, 2, 3, 4, 5, 6, 7 }，但不可能得到{ 3, 2, 1, 7, 5, 6, 4 }。

输入格式：

输入第一行给出 3 个不超过 1000 的正整数：M（堆栈最大容量）、N（入栈元素个数）、K（待检查的出栈序列个数）。最后 K 行，每行给出 N 个数字的出栈序列。所有同行数字以空格间隔。

输出格式：

对每一行出栈序列，如果其的确是有可能得到的合法序列，就在一行中输出YES，否则输出NO。

输入样例：

5 7 5
1 2 3 4 5 6 7
3 2 1 7 5 6 4
7 6 5 4 3 2 1
5 6 4 3 7 2 1
1 7 6 5 4 3 2

输出样例：

YES
NO
NO
YES
NO

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

---

题解：

 

/**
合法性：
    规律: --- 任意元素，比它序号小的元素序列必然是逆序的
    题目要求：容量最大为 M
思路：
    1.入栈第一个出栈的最大元素即为最大容量的元素: M
    2.用一个队列来存放出栈序列元素（对入栈待定合法的判断）
    3.num入栈数组里    if(a[num] <= 队头元素)    入栈    else    出栈

    4.若是此时的a[num] > 队头元素，后边的元素必然也比其大，不可能达到匹配，

        故可以直接退出，判定失败，不合法
    
*/

代码：

---

#include <iostream>
#include <stack>
#include <queue>
using namespace std;
int main()
{
    int m,n,k;
    cin >> m >> n >> k;
    int a[n];
    //将要操作的n个数字
    for(int i = 1; i <= n; i++)    a[i] = i;
    //进行k次循环判断
    while(k--)
    {
        stack<int> st;
        queue<int> q;   
        int x;
        //首先,进行此次的数字的录入队列
        for(int i = 0; i < n; i++) {
            cin >> x;
            q.push(x);
        }
        //其次进行主要操作
        bool flag = true;    //合法性判断
        int num = 1;    //入栈的元素序号
        //num数组里    if(a[num] <= 队头元素)    入栈    else    出栈
        while(st.size() < m && !q.empty())
        {
            if(a[num] <= q.front())    st.push(a[num++]);
            else //后边的数只会比当前数字更大，若是此数不成立，则后边的数也不可能匹配得到，故失败可以直接结束循环了
            { flag = false;    break; }
            //若匹配到了（队头和栈头相等了，则此元素合法满足，可以进行下次判定了）
            //先要满足不为空的判定才可进行条件判断，否则可能会出现段错误
            while(!q.empty() && !st.empty() && q.front() == st.top())
            {  
                q.pop();
                st.pop();
            }        
        }
    //退出循环了
        if(flag && q.empty())    cout << "YES\n";
        else cout << "NO\n";    
    }
    return 0;
}