【c/c++】堆栈操作合法性

617 阅读1分钟

设以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;
int main()
{
    int n,m;
    cin>>n>>m;
    while (n--){
        int flag=0;
        stack<int> T;
        char str[10000];
        cin>>str;
        int i=0;
        while (str[i]!='\0'){
                
            if(str[i]=='S'){
                if(T.size()==m)
                flag=1;
                T.push(1);
            }

            if(str[i]=='X'){
              if(T.empty())
                flag=1;
                T.pop();
            }
            i++;
            if(flag){
                cout<<"NO"<<endl;
                break;
			}
            
            }
            if(!flag && T.size()!=0)
            cout<<"NO"<<endl;
            if(!flag && T.size()==0)
            cout<<"YES"<<endl;
        }
    return 0;
}