【PTA数据结构 | C语言版】顺序栈的3个操作

本专栏持续输出数据结构题目集，欢迎订阅。

题目

请编写程序，将 n+1 个整数顺序压入容量为 n 的栈，随后执行 n+1 次取顶并出栈的操作。

输入格式：
输入首先在第一行给出正整数 n（≤10^4 ）；随后一行给出 n+1 个 int 范围内的整数，数字间以空格分隔。

输出格式：
将输入的n+1 个整数顺序压入容量为 n 的栈，随后执行 n+1 次取顶并出栈的操作，输出取出的元素的值，每行一个。

注意：当栈已满时，入栈操作应该不执行，并在一行中输出错误信息 错误：栈已满。；当栈为空时，取顶和出栈操作应该不执行，并在一行中输出错误信息 错误：栈为空。空栈取顶应返回 -1。

输入样例：
5
1 2 3 4 5 6

输出样例：
错误：栈已满。
5
4
3
2
1
错误：栈为空。
-1
错误：栈为空。

代码

#include <stdio.h>

#define MAX_SIZE 10000

int stack[MAX_SIZE];
int top = -1;

// 入栈操作
void push(int value, int n) {
    if (top >= n - 1) {
        printf("错误：栈已满。\n");
        return;
    }
    stack[++top] = value;
}

// 出栈操作
int pop() {
    if (top < 0) {
        printf("错误：栈为空。\n");
        return -1;
    }
    return stack[top--];
}

// 获取栈顶元素
int peek() {
    if (top < 0) {
        printf("错误：栈为空。\n");
        return -1;
    }
    return stack[top];
}

int main() {
    int n;
    scanf("%d", &n);
    
    // 压入n+1个元素
    for (int i = 0; i < n + 1; i++) {
        int num;
        scanf("%d", &num);
        push(num, n);
    }
    
    // 执行n+1次取顶和出栈
    for (int i = 0; i < n + 1; i++) {
        int top_val = peek();
        if (top_val != -1) {
            printf("%d\n", top_val);
        } else {
            printf("%d\n", -1);
        }
        pop();
    }
    
    return 0;
}