7-31 堆栈操作合法性 (20 分) PTA

7-31 堆栈操作合法性 (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 <iostream>
using namespace std;

int main(){
    int n,m;
    cin>>n>>m;
    for(int i=0;i<n;i++){
        string s;
        cin>>s;
        int sum=0;
        int flag = 0;
        for(int j=0;s[j]!='\0';j++){
            if(s[j]=='S'){
                sum++;
                if(sum>m){
                    cout<<"NO"<<endl;
                    flag = 1;
                    break;
                }
            } else{
                sum--;
                if(sum<0){
                    cout<<"NO"<<endl;
                    flag = 1;
                    break;
                }
            }
        }
        if(!flag){
            if(sum==0)
                cout<<"YES"<<endl;
            else
                cout<<"NO"<<endl;
        }
    }
    return 0;
}

算法思想：入栈表示+1，出栈表示-1，用sum来记录最终的数值。若中途出现sum<0 表示栈空，出栈失败，此时输出 NO。若中途sum值大于m，表示栈满，此时也输出 NO 。

若最终sum==0，且都不满足以上两个问题，此时输出YES。