数据结构(3)：顺序栈的表示和实现

/*  语言：C++                  编译环境：Visual C++6.0
顺序栈的表示和实现
*/
#include <iostream>
#define OK 1
#define ERROR 0
#define OVERFLOW -2
#define MAXSIZE 100
// Status是函数返回值类型，其值是函数结果状态代码
typedef int Status;
// 自定义数据类型别名
typedef int ElemType;
using namespace std;

// 顺序栈的存储结构
typedef struct
{
ElemType *base; // 栈底指针
ElemType *top;  // 栈顶指针
int stacksize;  // 栈可用的最大容量
}SqStack;

// 栈的初始化
Status InitStack(SqStack &S)
{   // 构造一个空栈
// 为顺序栈动态分配一个最大容量为MAXSIZE的数组空间
S.base = new ElemType[MAXSIZE];
if(!S.base) exit(OVERFLOW);     // 存储分配失败
S.top = S.base;                 // top初始为base，空栈
S.stacksize = MAXSIZE;      // stacksize置为栈的最大容量MAXSIZE
return OK;
}

// 顺序栈的入栈
Status Push(SqStack &S, ElemType e)
{   // 插入元素e为新的栈顶元素
// 栈满(S.top的地址值与S.base的地址值之差为顺序栈的长度)
if(S.top-S.base == MAXSIZE) return ERROR;
*S.top++ = e;                   // 元素e压入栈顶，栈顶指针加1
return OK;
}

// 顺序栈的出栈
Status Pop(SqStack &S, ElemType &e)
{   // 删除S的栈顶元素，用e返回其值
if(S.top == S.base) return ERROR;   // 栈空
e = *--S.top;                   // 栈顶指针减1，将栈顶元素赋给e
return OK;
}

// 取顺序栈的栈顶元素
ElemType GetTop(SqStack S)
{   // 返回S的栈顶元素，不修改栈顶指针
if(S.top != S.base)     // 栈非空
return *(S.top-1);  // 返回栈顶元素的值，栈顶指针不变
return ERROR;
}

int main()
{
SqStack S;
// 列表菜单
cout<<"1 InitStack"<<endl;
cout<<"2 Push"<<endl;
cout<<"3 Pop"<<endl;
cout<<"4 GetTop"<<endl;

int choice,e;
while(1)
{
cout<<"Input a choice: ";
cin>>choice;

// 选择
switch(choice)
{
case 1:
InitStack(S);
continue;
case 2:
Push(S,3);
continue;
case 3:
Pop(S,e);
continue;
case 4:
cout<<"Get: "<<GetTop(S)<<endl;
continue;
default:
cout<<"End..."<<endl;
break;
}
break;
}
return 0;
}