算法刷题笔记 模拟栈（C++实现）

题目描述

• 实现一个栈，栈初始为空，支持四种操作：
  • push x – 向栈顶插入一个数x；
  • pop – 从栈顶弹出一个数；
  • empty – 判断栈是否为空；
  • query – 查询栈顶元素。
• 现在要对栈进行M个操作，其中的每个操作3和操作4都要输出相应的结果。

输入格式

• 第一行包含整数M，表示操作次数。
• 接下来M行，每行包含一个操作命令，操作命令为push x，pop，empty，query中的一种。

输出格式

• 对于每个empty和query操作都要输出一个查询结果，每个结果占一行。
• 其中，empty操作的查询结果为YES或NO，query操作的查询结果为一个整数，表示栈顶元素的值。

数据范围

• 1 ≤ M ≤ 100000,
• 所有操作保证合法。

基本思路

栈的实现思路要远远比链表简单，因为每一个结点的值都不用管，只需要记录栈顶元素即可。同样，本题仍然是使用数组这种数据结构来模拟队列。

实现代码

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int stack[N];
int top = -1;

inline void push(void)
{
    int x;
    cin >> x;
    stack[++top] = x;
}

inline void pop(void)
{
    top--;
}

inline void empty(void)
{
    if(top == -1) cout << "YES" << endl;
    else cout << "NO" << endl;
}

inline void query(void)
{
    cout << stack[top] << endl;
}

int main(void)
{
    int M;
    cin >> M;
    for(int i = 0; i < M; ++i)
    {
        string operation;
        cin >> operation;
        if(operation == "push") push();
        else if(operation == "pop") pop();
        else if(operation == "empty") empty();
        else if(operation == "query") query();
    }
    return 0;
}