AcWing 828. 模拟栈

\(AcWing\) \(828\). 模拟栈

一、题目描述

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

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,1≤x≤10^9\)
所有操作保证合法。

输入样例：

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

输出样例：

5
5
YES
4
NO

二、理解与感悟

1、$tt$确定初始值是$0$,但增加时++$tt$, 就是实际上数组是从$1$开始的。

2、弹出就\(tt\)--,指针变了，但多余的数据不用清除，不碍事。

3、\(tt\)回到\(0\)，就是一个都没有了。

4、用数组模拟栈，比\(STL\)的\(stack\)方便、速度快，可以遍历到栈中每一个元素。

三、实现代码

#include <bits/stdc++.h>

using namespace std;
const int N = 100010;
int stk[N], tt;
string cmd;

int main() {
    int n;
    cin >> n;
    while (n--) {
        cin >> cmd;
        if (cmd == "push") {
            int x;
            cin >> x;
            stk[++tt] = x;
        } else if (cmd == "pop")
            tt--;
        else if (cmd == "query")
            printf("%d\n", stk[tt]);
        else if (cmd == "empty") {
            if (tt == 0) printf("YES\n");
            else printf("NO\n");
        }
    }
    return 0;
}