题目链接:https://www.acwing.com/problem/content/830/
题意:实现一个栈,栈初始为空,支持四种操作:
(1) “push x” – 向栈顶插入一个数x;
(2) “pop” – 从栈顶弹出一个数;
(3) “empty” – 判断栈是否为空;
(4) “query” – 查询栈顶元素。
现在要对栈进行M个操作,其中的每个操作3和操作4都要输出相应的结果。
数据范围
1≤M≤1e5,
1≤x≤1e9
所有操作保证合法。
输入样例:
10
push 5
query
push 6
pop
query
pop
empty
push 4
query
empty
输出样例:
5
5
YES
4
NO
思路:其实就是一个道理:栈遵循先进后出,而队列遵循先进先出。
代码实现:
#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,n) for(int i=a;i<=n;i++)
typedef long long ll;
const int INF=0x3f3f3f3f;
const int N = 1e5 + 5;
int m;
int stk[N], t;
int main()
{
cin >> m;
while (m -- ){
string op;
int x;
cin >> op;
if (op == "push"){
cin >> x;
stk[ ++ t] = x;
}
else if (op == "pop") t -- ;
else if (op == "empty")
cout << (t ? "NO" : "YES") << endl;
else cout << stk[t] << endl;
}
return 0;
}