PTA 堆栈操作合法性

堆栈操作合法性

题目要求：
假设以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

---

解题思路：
这题虽然是对堆栈的操作，但其实我们并没必要真的去写一个栈出来，只要去模拟这个过程就可以了。这里我先用了一个一维的字符串数组来接受输入，用sum表示栈区里的元素个数，开始sum初始化为0。然后扫描这个字符串，若字符为’S’，则代表入栈，这时只要sum ++就可以了，同理，字符为’X’时，sum –。在任何一个只要sum < 0或者sum > m，则用flag = 0表示错误，并跳出循环。最后整个字符串扫描完毕后，判断sum是否等于0，并且flag是否为0就可以看出堆栈操作是否合法了。没做完一组数据，记得初始化数据。

---

Code：

#include <stdio.h>
#include <string.h>

int main()
{
    int i, n, m, sum;
    char str[101] = {0};
    scanf("%d %d", &n, &m);
    while(n --) {
        int flag = 1;
        sum = 0;
        scanf("%s", str);
        for(i = 0; i < strlen(str); i ++) {
            if(str[i] == 'S') {
                sum ++;
                if(sum > m) {
                    flag = 0;
                    break;
                }
            }
            else if(str[i] == 'X'){
                sum --;
                if(sum < 0) {
                    flag = 0;
                    break;
                }
            }
        }
        if(flag == 0 || sum != 0) {
            printf("NO\n");
        }
        else {
            printf("YES\n");
        }
    }

    return 0;
}