3.9 堆栈操作合法性 (20分)

堆栈操作合法性

假设以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

题目链接：https://pintia.cn/problem-sets/434/problems/5891

AC代码：

#include <map>
#include <cmath>
#include <stack>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;

int main() {

	int n, m;
	cin >> n >> m;

	while (n-- > 0) {
		string s;
		int count = 0;

		cin >> s;
		int i;
		for (i = 0; i < s.size(); i++) {
			if (s[i] == 'S') {
				count++;
			}
			if (s[i] == 'X') {
				count--;
			}
			if (count<0 || count>m) {
				break;
			}
		}

		if (!count) {
			cout << "YES" << endl;
		}
		else {
			cout << "NO" << endl;
		}
	}

	//system("pause");

	return 0;
}