7-2 堆栈操作合法性 (20分) 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

#include<iostream>
using namespace std;
const int MaxSize=100;
int main()
{
    int n,maxx,k,flag=0,i;
    cin>>k>>maxx;
    string s;
    while(k--)
    {
        cin>>s;
        flag=0;
        for(i=0; i<s.length(); i++)
        {
            if(s[i]=='S')
            {
                flag++;
            }
            else
            {
                flag--;
            }
            if(flag<0||flag>maxx){
                cout<<"NO"<<endl;
                break;
            }
        }
        if(i==s.length())
        {
            if(flag==0)//要求最后状态也为空队列
                cout<<"YES"<<endl;
            else
            {
                cout<<"NO"<<endl;
            }
        }

    }
    return 0;
}