23.Acwing基础课第828题-简单-模拟栈

23.Acwing基础课第828题-简单-模拟栈

题目描述

实现一个栈，栈初始为空，支持四种操作：

1. push x – 向栈顶插入一个数 xx；
2. pop – 从栈顶弹出一个数；
3. empty – 判断栈是否为空；
4. query – 查询栈顶元素。

现在要对栈进行 MM 个操作，其中的每个操作 33 和操作 44 都要输出相应的结果。

输入格式

第一行包含整数 M，表示操作次数。

接下来 M 行，每行包含一个操作命令，操作命令为 push x，pop，empty，query 中的一种。

输出格式

对于每个 empty 和 query 操作都要输出一个查询结果，每个结果占一行。

其中，empty 操作的查询结果为 YES 或 NO，query 操作的查询结果为一个整数，表示栈顶元素的值。

数据范围

1≤M≤100000

1≤x≤10⁹

所有操作保证合法，即不会在栈为空的情况下进行 pop 或 query 操作。

输入样例：

10
push 5
query
push 6
pop
query
pop
empty
push 4
query
empty

输出样例：

5
5
YES
4
NO

代码：

#include <iostream>

using namespace std;

const int N = 100010;

int m;
int stk[N], tt;

// 1. 压栈：向栈顶插入x
void push(int x) 
{
    stk[++tt] = x;
}

// 2. 弹栈：删除栈顶元素（栈非空）
void pop() 
{
    tt--;
}

// 3. 查询栈顶元素（栈非空）
int top()
{
    return stk[tt];
}

// 4. 判断栈是否为空
bool empty() 
{
    cout << (tt ? "NO" : "YES") << endl;
    return (tt ? "NO" : "YES");
}

int main()
{
    cin >> m;
    while (m -- )
    {
        string op;
        int x;

        cin >> op;
        if (op == "push")
        {
            cin >> x;
            push(x);
        }
        else if (op == "pop") pop();
        else if (op == "empty") empty();
        else cout << top() << endl;
    }

    return 0;
}