7-23 栈操作的合法性-优快云博客

7-23 栈操作的合法性

作者 DS课程组

单位浙江大学

假设以S和X分别表示入栈和出栈操作。如果根据一个仅由S和X构成的序列，对一个空堆栈进行操作，相应操作均可行（如没有出现删除时栈空）且最后状态也是栈空，则称该序列是合法的堆栈操作序列。请编写程序，输入S和X序列，判断该序列是否合法。

输入格式:

输入第一行给出两个正整数 n 和 m，其中 n 是待测序列的个数，m（≤50）是堆栈的最大容量。随后 n 行，每行中给出一个仅由S和X构成的序列。序列保证不为空，且长度不超过100。

输出格式:

对每个序列，在一行中输出YES如果该序列是合法的堆栈操作序列，或NO如果不是。

输入样例：

4 10
SSSXXSXXSX
SSSXXSXXS
SSSSSSSSSSXSSXXXXXXXXXXX
SSSXXSXXX

输出样例：

YES
NO
NO
NO

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

栈限制

8192 KB

代码

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n, m;
    cin >> n >> m;
    while (n--)
    {
        stack<char> charBag;
        string operationStr;
        cin >> operationStr;

        bool ifOut = false;
        for (char opeChar : operationStr)
        {
            if (opeChar == 'S')
            {
                if (charBag.size() == m) // 超出的情况
                {
                    cout << "NO\n";
                    ifOut = true;
                    break;
                }
                charBag.push('S'); // 正常入栈
            }
            else if (opeChar == 'X')
            {
                if (charBag.empty()) // 已经为空还取则错
                {
                    cout << "NO\n";
                    ifOut = true;
                    break;
                }
                charBag.pop(); // 正常出栈
            }
        }
        if (!ifOut)                                         // 还没输出的话
            cout << ((charBag.empty()) ? "YES\n" : "NO\n"); // 空则YES，否则NO
    }
    return 0;
}