本文将简单归纳两道acwing上的有关栈的题目,第一题较简单,第二题理解起来可能会有些困难。
1.题目:acwing828
实现一个栈,栈初始为空,支持四种操作:
push x
– 向栈顶插入一个数 x;pop
– 从栈顶弹出一个数;empty
– 判断栈是否为空;query
– 查询栈顶元素。
现在要对栈进行 M个操作,其中的每个操作 33 和操作 44 都要输出相应的结果。
输入格式
第一行包含整数 M,表示操作次数。
接下来 M行,每行包含一个操作命令,操作命令为 push x
,pop
,empty
,query
中的一种。
输出格式
对于每个 empty
和 query
操作都要输出一个查询结果,每个结果占一行。
其中,empty
操作的查询结果为 YES
或 NO
,query
操作的查询结果为一个整数,表示栈顶元素的值。
数据范围
1≤M≤100000
1≤x≤10^9
所有操作保证合法。
#include <iostream>
using namespace std;
const int N = 100010;
int st[N];
int top = -1;
int n;
int main()
{
cin >> n;
while (n--)
{
string s;
cin >> s;
//栈顶所在索引往后移动一格,然后放入x。
if (s == "push")
{
int a;
cin >> a;
st[++top] = a;
}
//往前移动一格
if (s == "pop")
{
top--;
}
//返回栈顶元素
if (s == "query")
{
cout << st[top] << endl;
}
//大于等于 0 栈非空,小于 0 栈空
if (s == "empty")
{
cout << (top == -1 ? "YES" : "NO") << endl;
}
}
}
这道题看似很简单,但是实际上也很简单。
以top为索引值,检索整个栈。关于栈的结构,我们可以把它想象成几块砖从上到下垒上去。
但是接下来就要上强度了。
2.题目:acwing3302
给定一个表达式,其中运算符仅包含 +,-,*,/
(加 减 乘 整除),可能包含括号,请你求出表达式的最终值。
注意:
- 数据保证给定的表达式合法。
- 题目保证符号
-
只作为减号出现,不会作为负号出现,例如,-1+2
,(2+2)*(-(1+1)+2)
之类表达式均不会出现。 - 题目保证表达式中所有数字均为正整数。
- 题目保证表达式在中间计算过程以及结果中,均不超过 231−1231−1。
- 题目中的整除是指向 00 取整,也就是说对于大于 00 的结果向下取整,例如 5/3=15/3=1,对于小于 00 的结果向上取整,例如 5/(1−4)=−15/(1−4)=−1。
- C++和Java中的整除默认是向零取整;Python中的整除
//
默认向下取整,因此Python的eval()
函数中的整除也是向下取整,在本题中不能直接使用。
输入格式
共一行,为给定表达式。
输出格式
共一行,为表达式的结果。
数据范围
表达式的长度不超过 10^5。
#include<iostream>
#include<algorithm>
#include<stack>
#include<unordered_map>
using namespace std;
stack<int> num;
stack<char> op;
void eval() {
auto b = num.top(); num.pop();
auto a = num.top(); num.pop();
auto c = op.top(); op.pop();
int x = 0;
if (c == '+') x = a + b;
else if (c == '-') x = a - b;
else if (c == '*') x = a * b;
else x = a / b;
num.push(x);
}
int main() {
unordered_map<char, int> pr{ {'+',1},{'-',1},{'*',2},{'/',2} };
string str;
cin >> str;
for (int i = 0; i < str.size(); i++) {
auto c = str[i];
if (isdigit(c)) {
int x = 0, j = i;
while (j < str.size() && isdigit(str[j]))
x = x * 10 + str[j++] - '0';
i = j - 1;
num.push(x);
}
else if (c == '(') op.push(c);
else if (c == ')') {
while (op.top() != '(') eval();
op.pop();
}
else {
while (op.size() && pr[op.top()] >= pr[c]) eval();
op.push(c);
}
}
while (op.size()) eval();
cout << num.top() << endl;
return 0;
}
我们选择读入一个字符串类型的变量,其中包括数字、括号以及运算符,处理起来需要综合考虑。
主要的思想就是分而治之,把把运算符两侧的运算看成最终的数,表面上的数字就是树的分支,从分支入手,一步步计算出来。
1.#include<stack>,表示调用stack的头文件,可以直接使用top,pop函数。#include<unordered_map>,表示调用哈希表的头文件,可以构建起一个哈希表,实际上这就是运算符优先级的哈希表。
2.eval函数,从num栈中取出两个操作数,从op栈中取出一个操作数,根据op栈中的操作数执行相应的运算,再把结果放入num栈中。本质上就是在做四则运算。
3.1main函数中,我们使用了哈希表存储操作符和它的优先级,加减优先级为1,乘除优先级为2
3.2接下来输入str字符串,我们用c来读取(auto会自动判断类型,不用担心int和char),接下来判断c是否为数字字符(采用的是isdigit函数)。如果c为数字字符,那么用x来存储数字,循环执行判断;如果c是左括号,那么把它压入op栈内;如果c是右括号,如果op栈内不是左括号,那么就不断执行运算直至c为左括号,然后把左括号删除;当然c除了数字,括号外还可能是运算符,那么就直接把他们压入op栈内。
4.最后,我们把各个小部分计算完后进行最终的计算,然后输出最终结果。