堆栈的使用

题目描述

堆栈是一种基本的数据结构。堆栈具有两种基本操作方式，push 和 pop。Push一个值会将其压入栈顶，而 pop 则会将栈顶的值弹出。现在我们就来验证一下堆栈的使用。

输入描述

对于每组测试数据，第一行是一个正整数 n，0<n<=10000(n=0 结束)。而后的 n 行，每行的第一个字符可能是'P’或者'O’或者'A’；如果是'P’，后面还会跟着一个整数，表示把这个数据压入堆栈；如果是'O’，表示将栈顶的值 pop 出来，如果堆栈中没有元素时，忽略本次操作；如果是'A’，表示询问当前栈顶的值，如果当时栈为空，则输出'E'。堆栈开始为空。

输出描述

 对于每组测试数据，根据其中的命令字符来处理堆栈；并对所有的'A’操作，输出当时栈顶的值，每个占据一行，如果当时栈为空，则输出'E’。当每组测试数据完成后，输出一个空行。

输入样例

5
P 75
O
O
P 60
A
7
A
O
P 73
P 49
A
O
P 3
0

输出样例

60

E
49

 

Code:

#include <stdio.h>
#include <stdlib.h>

#define STACK_SIZE 1

typedef struct
{
    int *base;//栈底
    int *top;//栈顶
    int stacksize;
}Stack;//堆栈

void InitStack(Stack *S);//创建堆栈
void Push(Stack *S, int elem);//向堆栈中存入数据
void GetTop(Stack *S);//输出栈顶数据
int Pop(Stack *S);//弹出栈顶数据

int main()
{
    char c[3];
    int elem;
    int n;
    int i;

    while (scanf("%d", &n) != EOF)//多组数据
    {
        Stack S;
        InitStack(&S);
        for (i = 0; i < n; i++)
        {
            scanf("%s", c);//相关操作
            if (c[0] == 'P')
            {
                scanf("%d", &elem);
                Push(&S, elem);
            }
            else if (c[0] == 'O')
            {
                Pop(&S);
            }
            else if (c[0] == 'A')
            {
                GetTop(&S);
            }
        }
        printf("\n");
    }
    return 0;
}

void InitStack(Stack *S)
{
    S->base = (int*)malloc(STACK_SIZE * sizeof(int));//分配储存
    if (!S->base)
    {
        exit(0);
    }
    S->top = S->base;
    S->stacksize = STACK_SIZE;
}

void Push(Stack *S, int elem)
{
    if (S->top - S->base >= S->stacksize)
    {
        S->base = (int*)realloc(S->base, (S->stacksize + 1) * sizeof(int));
        S->top = S->base + S->stacksize;
        S->stacksize++;
    }
    if (!S->base)
    {
        exit(0);
    }
    *(S->top++) = elem;
}

void GetTop(Stack *S)
{
    if (S->top == S->base)
    {
        printf("E\n");
    }
    else
    {
        printf("%d\n", *(S->top - 1));
    }
}

int Pop(Stack *S)
{
    if (S->top == S->base)
    {
        return 1;
    }
    else
    {
         --S->top;
    }
}