问题 C: 第3关：入栈和出栈的基本操作

题目描述

本关任务：输入一个整数序列a1,a2,a3...,an。当ai不等于-1时将ai进栈；当ai=-1时，输出栈顶元素并将其出栈。

输入描述

多组数据，每组数据有两行，第一行为序列的长度n，第二行为n个整数，整数之间用空格分隔。当n=0时输入结束。

输出描述

对于每一组数据输出若干行。每行为相应的出栈元素。当出栈异常时，输出“POP ERROR”并结束本组数据的输出。

代码实现 

#include<iostream>

using namespace std;
#define MAXSIZE 100
#define OK 1
#define ERROR 0
#define OVERFLOW - 2
typedef struct {
  int * base;
  int * top;
  int stacksize;
}
SqStack;

int InitSqStack(SqStack & S) { //栈的初始化
  S.base = new int[MAXSIZE];
  if (!S.base) return OVERFLOW;
  S.top = S.base;
  S.stacksize = MAXSIZE;
  return OK;
}

int Push(SqStack & S, int e) { //入栈
  if (S.top - S.base == S.stacksize) //栈满
    return ERROR;
  * S.top = e;
  S.top++;
  return OK;
}

int Pop(SqStack & S) { //出栈
  if (S.top == S.base) //栈空
    return ERROR;
  S.top--;
  return OK;
}

int GetTop(SqStack S) { //取栈顶元素
  if (S.top == S.base) //栈空
    return ERROR;
  return * (S.top - 1);
}

void InOutS(SqStack &S, int a[], int n) {
    for (int i = 0;i < n; ++i){
        if(a[i] != -1){
            Push(S,a[i]);
        }
        else
        {
            if(GetTop(S)){
                cout << GetTop(S) << endl;
                 Pop(S);
            }
            else{
                cout << "POP ERROR" << endl;
                 break;
            }
        }
    }
}

int main() {
  int n;
  while (cin >> n) {
    if (n == 0) break;
    SqStack S;
    InitSqStack(S);
    int a[n];
    for (int i = 0; i < n; i++) cin >> a[i]; //整数序列
    InOutS(S, a, n);
  }
  return 0;
}