PTA数据结构之 堆栈操作合法性

假设以S和X分别表示入栈和出栈操作。如果根据一个仅由S和X构成的序列，对一个空堆栈进行操作，相应操作均可行（如没有出现删除时栈空）且最后状态也是栈空，则称该序列是合法的堆栈操作序列。请编写程序，输入S和X序列，判断该序列是否合法。

输入格式:

输入第一行给出两个正整数N和M，其中N是待测序列的个数，M（$\le 50$）是堆栈的最大容量。随后N行，每行中给出一个仅由S和X构成的序列。序列保证不为空，且长度不超过100。

输出格式:

对每个序列，在一行中输出YES如果该序列是合法的堆栈操作序列，或NO如果不是。

输入样例：

4 10
SSSXXSXXSX
SSSXXSXXS
SSSSSSSSSSXSSXXXXXXXXXXX
SSSXXSXXX

输出样例：

YES
NO
NO
NO

代码实现：

#include<stdio.h>
#include<stdlib.h>
#include<malloc.h>
#include<iostream>
using namespace std;
#define OK 1
#define ERROR 0
#define TRUE 1
#define FALSE 0
#define OVERFLOW -2
#define STACK_INIT_SIZE 100
#define STACKINCREMENT 10
typedef char SElemType;
typedef int Status;
typedef struct
{
    SElemType *base;
    SElemType *top;
    int stacksize;
}SqStack;
Status Init_S(SqStack &S)
{
    S.base=(SElemType *)malloc(sizeof(SElemType)*STACK_INIT_SIZE);
    if(!S.base) exit(OVERFLOW);
    S.top=S.base;//空栈
    S.stacksize=STACK_INIT_SIZE;
    return OK;
}
Status Push_Stack(SqStack &S,SElemType e)//ruzhan
{
    if(S.top-S.base==S.stacksize)//如果栈满
    {
        S.base=(SElemType *)realloc(S.base,(S.stacksize+STACKINCREMENT)*sizeof(SElemType));
        if(!S.base) exit(OVERFLOW);
        S.top=S.base+S.stacksize;//使S.top重新指向栈顶
        S.stacksize+=STACKINCREMENT;
    }
    *S.top=e;
    S.top++;
    return OK;
}
Status Empty_Stack(SqStack &S)
{
    if(S.top-S.base==0)
        return TRUE;
    else
        return FALSE;
}
Status Pop_Stack(SqStack &S)
{
    if(S.top-S.base==0)
        return ERROR;
    S.top--;
    return OK;
}
Status Destory_S(SqStack &S)
{
    free(S.base);
    S.base=NULL;
    S.top=NULL;
    S.stacksize=0;
    return OK;
}
int JudgeStacksize(SqStack &S)
{
    return (S.top-S.base);
}
int main()
{
    int m,n,geshu;
    string e;
    cin>>m>>n;
    for(int i=0;i<m;i++)
    {
        geshu=0;
        SqStack S;
        Init_S(S);
        cin>>e;
        for(int j=0;j<e.length();j++)
        {
                if(e[j]=='S')
                {
                    if(!Push_Stack(S,e[j]))//判断不合法情况
                    {
                        geshu=1;
                        break;
                    }
                }
                else if(e[j]=='X')
                {
                    if(!Pop_Stack(S))//判断不合法情况
                    {
                        geshu=1;
                    }
                }
                if(JudgeStacksize(S)>n)
        {
            geshu=1;
        }

        }

        if(!Empty_Stack(S))
        {
            geshu=1;
        }
        if(geshu==0)
        {
            cout<<"YES";
        }
        else
        {
            cout<<"NO";
        }

        if(i!=m-1)
            cout<<endl;
        Destory_S(S);

    }

    return 0;

}