题目描述

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

1. push x – 向栈顶插入一个数$x$；
2. pop – 从栈顶弹出一个数；
3. empty – 判断栈是否为空；
4. 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

代码&思路

#include<iostream>
using namespace std;
const int N=100010;
int stk[N],tt;
int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        int x;
        string op;
        cin>>op;
        if(op=="push")
        {
            cin>>x;
            stk[++tt]=x;//头结点++，就代表多放入了一个元素
        }
        if(op=="pop")
        {
            tt--;//头结点--，就代表少放入一个元素，也就是弹出
        }
        if(op=="empty")
        {
            if(!tt) cout<<"YES"<<endl;//如果tt=0，就说明栈里面什么都没有，就输出"YES"
            else cout<<"NO"<<endl;//反之输出"NO"
        }
        if(op=="query")
        {
            cout<<stk[tt]<<endl;//tt是头结点，所以返回头结点直接输出tt的值即可
        }
    }
    return 0;
}