PTA（java）堆栈操作合法性

作者: DS课程组

单位: 浙江大学

时间限制: 400ms

内存限制: 64MB

代码长度限制: 16KB

7-14 堆栈操作合法性（20 分）

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

输入格式:

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

输出格式:

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

输入样例：

4 10
SSSXXSXXSX
SSSXXSXXS
SSSSSSSSSSXSSXXXXXXXXXXX
SSSXXSXXX

输出样例：

YES
NO
NO
NO

代码：

import java.util.Scanner;

public class Main {

public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int N=scanner.nextInt();
int M=scanner.nextInt();
if(M<=50)
{
String str[] = new String[N];
for(int i=0;i<N;i++)
{
str[i]=scanner.next();
}
scanner.close();
for(int i=0;i<N;i++)
{
if(str[i].length()>100) {System.out.println("NO");continue;}
int S=0,X=0;
for(int j=0;j<str[i].length();j++)
{
if(str[i].charAt(j)=='S') {S++;}
if(str[i].charAt(j)=='X') {X++;}
}
if(S!=X||S>M) {System.out.println("NO");}
else
{
for(int j=0;j<str[i].length();j++)
{
S=0;X=0;
for(int l=0;l<=j;l++)
{
if(str[i].charAt(l)=='S') {S++;}
if(str[i].charAt(l)=='X') {X++;}
}
if(S<X) {System.out.println("NO");break;}
if(j==(str[i].length()-1)&&S==X) {System.out.println("YES");break;}
}
}
}
}
}
}