编程题：栈操作的合法性

作者 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

答案：

#include <stdio.h>

int main() {
    int n, m, i, length, flag;
    scanf("%d %d", &n, &m);
    getchar(); // 读取并丢弃剩余的换行符

    for (i = 0; i < n; i++) {
        length = 0;
        flag = 0;
        char ch;

        while ((ch = getchar()) != '\n') {
            if (ch == 'S') {
                length++;
                if (length > m) flag = 1;
            } else if (ch == 'X') {
                length--;
                if (length < 0) flag = 1;
            }
        }

        if (length == 0 && flag == 0)
            printf("YES\n");
        else
            printf("NO\n");
    }

    return 0;
}