一、算法描述
本篇文章讲述的数据结构是,栈,数组模拟栈。
栈的结构相信大家应该很清楚了,特点就是先进后出,只能在栈顶操作,栈底不能操作。
//用数组模拟的栈定义如下:int tt;int st[N];/*tt表示栈顶(我习惯于表示栈顶的下一个位置,可以根据个人习惯来修改)st[N]表示栈*/
- 栈不是很难理解,相比于链表要简单很多,后面的队列和栈一样不是很难理解。
接下来介绍栈的各种操作:初始化操作:
void init(){ tt = 0;}
- 看个人习惯,我习惯于表示栈顶元素的下一个位置,如果是表示栈顶初始化应修改为 \(tt = -1\);
向栈中压入元素:
void push(int x){ st[tt ++ ] = x; //st[++ tt ] = x;}
- 以上两种写法的区别就在于,第一种是先存储了 \(x\) ,\(tt\) 再加;而第二种是 \(tt\) 先加再存储 \(x\)。
- 主要就是个人习惯问题。
弹出栈顶元素:
void pop(){ -- tt;}
- 栈只能在栈顶操作,所以直接让栈顶指针减一即可。
判空:
bool empty(){ return tt == 0;}
- 直接判断栈顶指针是否为 \(0\) 即可。
- 同样另一种情况就是 \(tt == -1\) 即可。
查询栈顶元素:
int query(){ return st[tt - 1];}
- 直接返回栈顶元素即可。
- 另一种写法就是 \(st[tt]\) 即可。
栈和队列的问题都不是很难,主要还是要多应用,熟练掌握这些数据结构。
二、题目描述
实现一个栈,栈初始为空,支持四种操作:
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\)
所有操作保证合法。
输入样例:
10push 5querypush 6popquerypopemptypush 4queryempty
输出样例:
55YES4NO
三、题目来源 AcWing算法基础课-828.模拟栈四、源代码
#include #include using namespace std;const int N = 100010;int tt;int st[N];void init(){ tt = 0;}void push(int x){ st[tt ++ ] = x;}void pop(){ -- tt;}bool empty(){ return tt == 0;}int query(){ return st[tt - 1];}int main(){ int m; cin >> m; init(); while (m -- ) { char s[10]; cin >> s; int x; if (!strcmp(s, "push")) { cin >> x; push(x); } else if (!strcmp(s, "pop")) { pop(); } else if (!strcmp(s, "empty")) { if (empty()) puts("YES"); else puts("NO"); } else { cout << query() << endl; } } return 0;}