7-2 堆栈操作合法性 (20 分) C++

假设以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>
#include<string>
#include<stack>
using namespace std;
int main()
{
    string ch;
    stack<int> chr;
    int n,m;
    cin>>n>>m;
    while(n--)
    {
        if(m==0)
        {
            if(n!=0) cout<<"NO"<<endl;
            else cout<<"NO";
        }
        else
        {
            while(!empty(chr))
            {
                chr.pop();
            }
            int flag=0;
            cin>>ch;
            for(int i=0;i<size(ch);i++)
            {
                if(ch[i]=='X'&&empty(chr))
                {
                    flag=1;
                    break;
                }
                else if(ch[i]=='S'&&size(chr)<m)
                {
                    chr.push(1);
                }
                else if(ch[i]=='S'&&size(chr)>=m)
                {
                    flag=1;
                    break;
                }
                else if(ch[i]=='X'&&!empty(chr))
                {
                    chr.pop();
                }
            }
            if(!empty(chr)||flag==1)
            {
                if(n!=0)
                    cout<<"NO"<<endl;
                else cout<<"NO";
            }
            else if(empty(chr)&&flag==0){
                if(n!=0)
                    cout<<"YES"<<endl;
                else cout<<"YES";
            }
        }
    }
}
上一篇:python – 将大量数据写入stdin


下一篇:L1-009. N个数求和