7-2 堆栈操作合法性 (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
代码如下:
#include <bits/stdc++.h>
using namespace std;
char s[1000];
int len;
int N, M;
void solve() {
int cnt = 0;
for(int i = 0 ; i < len; ++i) {
if(s[i] == 'S') cnt++;
else if(s[i] == 'X') cnt--;
if(cnt > M || cnt < 0) { puts("NO"); return; }
}
if(cnt != 0) { puts("NO"); return; }
puts("YES");
}
int main()
{
cin >> N >> M;
for(int i = 0; i < N; i++)
{
scanf("%s", s);
len = strlen(s);
solve();
}
}