【LeetCode专题#基本计算器】基本计算器I，图解中序表达式转逆波兰表达式，太难了
基本计算器
https://leetcode.cn/problems/basic-calculator/?envType=list&envId=cKNEfNsF
给你一个字符串表达式 s ，请你实现一个基本计算器来计算并返回它的值。
注意:不允许使用任何将字符串作为数学表达式计算的内置函数，比如 eval() 。
示例 1：
输入：s = "1 + 1"
输出：2
示例 2：
输入：s = " 2-1 + 2 "
输出：3
示例 3：
输入：s = "(1+(4+5+2)-3)+(6+8)"
输出：23
提示：
1 <= s.length <= 3 * 105
s 由数字、'+'、'-'、'('、')'、和 ' ' 组成
s 表示一个有效的表达式
'+' 不能用作一元运算(例如， "+1" 和 "+(2 + 3)" 无效)
'-' 可以用作一元运算(即 "-1" 和 "-(2 + 3)" 是有效的)
输入中不存在两个连续的操作符
每个数字和运行的计算将适合于一个有符号的 32位 整数
思路
本题由两种方法解决，一种是单调栈，另一种是逆波兰表达式（主要考虑逆波兰）
使用逆波兰表达式的话大概思路如下：
创建一个操作符栈op和一个结果栈res并将字符串s（s为中序表达式）的第一个字符设置为当前字符
先将输入的中序字符串s中的空格处理掉，然后按以下规则从左到右依次遍历中序字符串s的每个字符
a. 如果当前字符是数字，则直接压入结果栈中；
b. 如果当前字符是操作符，则分情况讨论：
i. 如果当前操作符优先级低于或等于操作符栈顶元素，则将操作符栈顶元素弹出并加入结果栈中，重复此步骤直到操作符栈为空或 者操作符的优先级高于栈顶元素为止。然后将操作符压入操作符栈中。
ii. 如果当前操作符优先级高于操作符栈顶元素，则直接将操作符压入操作符栈中。
c. 如果当前字符是左括号，则将其压入操作符栈中。
d. 如果当前字符是右括号，则不断地将操作符栈顶元素弹出并加入结果栈中，直到遇到左括号为止。
当中序表达式遍历完毕后，如果操作符栈中还有剩余元素，则将其全部弹出并加入结果栈中。 最后，结果栈中的元素就是逆波兰表达式。
假设我们有中序表达式 "3 + 4 * (2 - 1)"，现在来演示如何将其转换为逆波兰表达式：
首先先去空格，得到"3+4*(2-1)"
然后，创建操作符栈和结果栈，并将表达式的第一个字符 "3" 设置为当前字符。
接着，遍历表达式中的下一个字符 "+":
因为 "+" 是一个操作符，所以将其与操作符栈顶元素比较。此时操作符栈为空，因此直接将 "+" 压入操作符栈中。
下一个字符是数字 "4"，因此将其直接压入结果栈中。
下一个字符是操作符 "*"，再次将其与操作符栈顶元素比较。因为 "*" 的优先级高于 "+"，所以直接将 "*" 压入操作符栈中。
下一个字符是左括号 "("，将其压入操作符栈中。
下一个字符是数字 "2"，将其压入结果栈中。
下一个字符是操作符 "-"，将其压入操作符栈中。
下一个字符是数字 "1"，将其压入结果栈中。
下一个字符是右括号 ")"，此时需要不断地将操作符栈顶元素弹出并加入结果栈中，直到遇到左括号为止。因此，弹出 "-"、"1" 和 "(" 并将它们依次加入结果栈中。
现在已经遍历完了全部字符，但是操作符栈中还有剩余元素"*"和"+"，因此将其弹出并加入结果栈中。
最后，结果栈中
的元素就是逆波兰表达式："3421-*+"
这个逆波兰表达式可以通过栈来求值。
图解如下：
遍历中序字符串s，将操作符按优先级压入操作符栈op，将数字按遍历顺序（从左到右）压入结果栈res
直到遇到右括号，这时，不断遍历操作符栈op，将操作符元素压入结果栈res
直到遇到左括号结束
此时，如果op内还有剩余操作符元素，将其依次弹出并加入res
得到结果"3421-*+"
逆波兰表达式实现思路
代码实现中，我们需要写四个函数：去除空格的函数removeBlack、逆波兰表达式转换函数toRPN、逆波兰表达式求解函数getRes以及主函数calculate
class Solution {
private:
//有一个移除空格的函数
string removeBlack(string s){
string res = "";
for(char c : s){
if(c != ' ') res += c;
}
return res;
}
//将表达式分割为后缀表达式
void toRPN(string s){
...
}
//将表达式分割为后缀表达式
void getRes(){
...
}
public:
int calculate(string s) {
}
};
getRes()函数再计算逆波兰表达式的结果时，需要一个vector
这里还添加了一个priority函数用于返回操作符的优先级
因此，toRPN的返回值应该是vector
class Solution {
private:
//有一个移除空格的函数
string removeBlack(string s){
string res = "";
for(char c : s){
if(c != ' ') res += c;
}
return res;
}
// 定义运算符优先级
int priority(char op) {
if (op == '+' || op == '-') return 1;
if (op == '*' || op == '/') return 2;
return 0;
}
//将中序表达式分割为后缀表达式
vector<string> toRPN(string s){
stack<string> op;//操作符栈
vector<string> res;//结果栈
string num = "";//记录多位数字
for(char c : s){////遍历当前输入的中序字符串s,判断当前字符的类型
if(isdigit(c)){//当前字符c为数字
num += c;//为了防止当前数字有多位数，先用num收集
}else if(is_operator(c)){//当前字符c为操作符
if(!num.empty()){//证明num已经将数字收集完毕，压入结果栈
res.push_back(num);
num = "";
}
//处理操作符逻辑，压入op
//当前op中有操作符元素，比较两者优先级
//【如果op.top() == "("则说明现在之前遇到了")",现在正在弹出"("之前的所有操作符】
//以下情况表示栈顶运算符的优先级大于等于当前读入的运算符c的优先级
while(!op.empty() && op.top() != "(" && priority(op.top()) >= priority(string(1, c))){
res.push_back(op.top());//将优先级高的先压入结果栈res
op.pop();//然后弹掉
}
op.push(string(1, c));//当前读入的运算符优先级大于等于栈顶运算符
}else if(c == '('){//当前字符c为左括号
if(!num.empty()){//证明num已经将数字收集完毕，压入结果栈
res.push_back(num);
num = "";
}
op.push("(");
}else if(c == ')'){//当前字符c为右括号
if(!num.empty()){//证明num已经将数字收集完毕，压入结果栈
res.push_back(num);
num = "";
}
while(op.top() != "(" && !op.empty()){//不断弹出操作符栈顶元素加入到结果栈res中
res.push_back(op.top());
op.pop();
}
op.pop();//多弹一次取出左括号
}
}//遍历处理完毕，如果num和操作符栈还有剩的元素，先将num压入res再压op
if(!num.empty()){//如果还有数字未入结果栈，则加入
res.push_back(num);
}
while(!op.empty()){//将剩余的操作符入结果栈
res.push_back(op.top());
op.pop();
}
return res;
}
//计算逆波兰表达式，LeetCode.150
int getRes(vector<string>& tokens){
}
public:
int calculate(string s) {
}
};
第一版代码
基于LeetCode.150的知识补充完getRes函数后我们得到了第一版代码，是不是很爽？示例用例都ac
class Solution {
private:
//有一个移除空格的函数 //只是将原字符串的空格去除后，生成了一个新的字符串，但并没有将原字符串进行修改,错误
string removeBlack(string& s){
string res = "";
for(char c : s){
if(c != ' ') res += c;
}
return res;
}
// 定义运算符优先级
int priority(string op) {
if (op == "+" || op == "-") return 1;
if (op == "*" || op == "/") return 2;
return 0;
}
bool is_operator(char c) {//判断当前元素是否为操作符
switch(c) {
case '+':
case '-':
case '*':
case '/':
return true;
default:
return false;
}
}
//将中序表达式分割为后缀表达式
vector<string> toRPN(string s){
stack<string> op;//操作符栈
vector<string> res;//结果栈
string num = "";//记录多位数字
for(char c : s){////遍历当前输入的中序字符串s,判断当前字符的类型
if(isdigit(c)){//当前字符c为数字
num += c;//为了防止当前数字有多位数，先用num收集
}else if(is_operator(c)){//当前字符c为操作符
if(!num.empty()){//证明num已经将数字收集完毕，压入结果栈
res.push_back(num);
num = "";
}
//处理操作符逻辑，压入op
//当前op中有操作符元素，比较两者优先级
//【如果op.top() == "("则说明现在之前遇到了")",现在正在弹出"("之前的所有操作符】
//以下情况表示栈顶运算符的优先级大于等于当前读入的运算符c的优先级
while(!op.empty() && op.top() != "(" && priority(op.top()) >= priority(string(1, c))){
res.push_back(op.top());//将优先级高的先压入结果栈res
op.pop();//然后弹掉
}
op.push(string(1, c));//当前读入的运算符优先级大于等于栈顶运算符
}else if(c == '('){//当前字符c为左括号
if(!num.empty()){//证明num已经将数字收集完毕，压入结果栈
res.push_back(num);
num = "";
}
op.push("(");
}else if(c == ')'){//当前字符c为右括号
if(!num.empty()){//证明num已经将数字收集完毕，压入结果栈
res.push_back(num);
num = "";
}
while(op.top() != "(" && !op.empty()){//不断弹出操作符栈顶元素加入到结果栈res中
res.push_back(op.top());
op.pop();
}
op.pop();//多弹一次取出左括号
}
}//遍历处理完毕，如果num和操作符栈还有剩的元素，先将num压入res再压op
if(!num.empty()){//如果还有数字未入结果栈，则加入
res.push_back(num);
}
while(!op.empty()){//将剩余的操作符入结果栈
res.push_back(op.top());
op.pop();
}
return res;
}
//计算逆波兰表达式，LeetCode.150
int getRes(vector<string>& tokens){
stack<long long> st;
//遍历字符串
for(int i = 0; i < tokens.size(); ++i){
if(tokens[i] == "+"|| tokens[i] == "-" || tokens[i] == "*" || tokens[i] == "/"){//遇到运算符
//取两个数
long long num1 = st.top();
st.pop();
long long num2 = st.top();
st.pop();
//判断运算符，做相应计算并压栈//注意计算顺序，num2[运算符]num1
if(tokens[i] == "+") st.push(num2 + num1);
if(tokens[i] == "-") st.push(num2 - num1);
if(tokens[i] == "*") st.push(num2 * num1);
if(tokens[i] == "/") st.push(num2 / num1);
}else{//遇到数字
//转为整型，压栈
st.push(stoll(tokens[i]));
}
}
int res = st.top();
st.pop();//内存回收
return res;
}
public:
int calculate(string s) {
string noBlackIn_s = removeBlack(s);
vector<string> rpn4s = toRPN(noBlackIn_s);
return getRes(rpn4s);
}
};
坑
但是，上述代码在测试用例为"1-( -2)"时报错了（咬牙切齿）
Line 171: Char 16: runtime error: reference binding to misaligned address 0xbebebebebebec0b6 for type 'long long', which requires 8 byte alignment (stl_deque.h)
0xbebebebebebec0b6: note: pointer points here
<memory cannot be printed>
SUMMARY: UndefinedBehaviorSanitizer: undefined-behavior /usr/bin/../lib/gcc/x86_64-linux-gnu/9/../../../../include/c++/9/bits/stl_deque.h:180:16
原因大概是因为按照我们之前的逻辑写的toRPN函数没有考虑到有负数的情况。。。
怎么办，改呗
第二版代码
考虑了负号，但是测试用例仍然只通过24/44
class Solution {
private:
string removeBlack(string& s){
string res = "";
int i = 0, n = s.size();
while(i < n){
if(s[i] == ' ') ++i;
else if(s[i] == '-'){
// 跳过空格
while(i < n && s[i] == ' ') ++i;
// 判断减号前是否有数字或左括号
if(i == 0 || s[i-1] == '('){
res += '0';
}
// 将减号复制到新字符串中
res += '-';
++i;
// 将减号后面的数字复制到新字符串中
while(i < n && isdigit(s[i])){
res += s[i];
++i;
}
// 在数字后面添加空格，以便后续处理
res += ' ';
}else{
// 复制其他字符到新字符串中
res += s[i];
++i;
}
}
return res;
}
int priority(string op) {
if (op == "+" || op == "-") return 1;
if (op == "*" || op == "/") return 2;
return 0;
}
bool is_operator(char c) {
switch(c) {
case '+':
case '-':
case '*':
case '/':
return true;
default:
return false;
}
}
vector<string> toRPN(string s){
stack<string> op;
vector<string> res;
string num = "";
for(int i = 0; i < s.size(); ++i){
char c = s[i];
if(isdigit(c)){
num += c;
}else if(is_operator(c)){
if(!num.empty()){
res.push_back(num);
num = "";
}
while(!op.empty() && op.top() != "(" && priority(op.top()) >= priority(string(1, c))){
res.push_back(op.top());
op.pop();
}
op.push(string(1, c));
}else if(c == '('){
if(!num.empty()){
res.push_back(num);
num = "";
}
op.push("(");
}else if(c == ')'){
if(!num.empty()){
res.push_back(num);
num = "";
}
while(op.top() != "(" && !op.empty()){
res.push_back(op.top());
op.pop();
}
op.pop();
}else if(c == ' '){ // 处理空格
if(!num.empty()){
res.push_back(num);
num = "";
}
// 如果前一个字符不是减号，则将空格添加到新字符串中
if(i > 0 && s[i-1] != '-'){
res.push_back(" ");
}
}
}
if(!num.empty()){
res.push_back(num);
}
while(!op.empty()){
res.push_back(op.top());
op.pop();
}
return res;
}
int getRes(vector<string>& tokens){
stack<int> st;
for(int i = 0; i < tokens.size(); ++i){
if(tokens[i] == "+"|| tokens[i] == "-" || tokens[i] == "*" || tokens[i] == "/"){
int num1 = st.top();
st.pop();
int num2 = st.top();
st.pop();
if(tokens[i] == "+") st.push(num2 + num1);
if(tokens[i] == "-") st.push(num2 - num1);
if(tokens[i] == "*") st.push(num2 * num1);
if(tokens[i] == "/") st.push(num2 / num1);
}else{
// st.push(stoi(tokens[i]));
stringstream ss(tokens[i]);
int num;
ss >> num;
st.push(num);
}
}
int res = st.top();
st.pop();
return res;
}
public:
int calculate(string s) {
string noBlackIn_s = removeBlack(s);
vector<string> rpn4s = toRPN(noBlackIn_s);
return getRes(rpn4s);
}
};
放弃逆波兰表达式的写法了
改用栈吧，下面贴一个论坛老哥的解法的分析，我修不动原来的代码了
class Solution {
public:
// 去除空格
string removeBlank(string s) {
string res = "";
for(char c:s) {
if(c!=' ') res += c;
}
return res;
}
// 将中缀表达式转化为后缀表达式
queue<string> getToken(string s) {
s = removeBlank(s);
string push_src = "";
queue<string> res;
bool pre = true;
for(char c:s) {
//判断是不是单目运算符 使用$代替单目运算符
if(c == '-' && pre) {
if(push_src!="") {
res.push(push_src);
push_src = "";
}
res.push("$");
}else if(c == '+' || c=='-' || c=='*' || c=='/' || c=='(' || c==')'||c=='#') {
if(c!=')')pre = true;
if(push_src!="") {
res.push(push_src);
push_src = "";
}
res.push(string("")+c);
}else{
pre = false;
push_src += c;
}
}
if(push_src!="") {
res.push(push_src);
push_src = "";
}
return res;
}
// 给定一个后缀表达式，求其值
int calculate(string s) {
queue<string> in = getToken(s+"#"); // #表示计算结束
map<string,int> isp = {
{"#",0},{"(",1},{"*",5},{"/",5},{"+",3},{"-",3},{")",8},{"$",7}
};
map<string,int> icp = {
{"#",0},{"(",8},{"*",4},{"/",4},{"+",2},{"-",2},{")",1},{"$",6}
};
queue<string> out; // 存储后缀表达式
stack<string> stk; // 操作符栈
stk.push("#");
string ch = in.front(); // 取队首元素
in.pop();
while(stk.top()!="#"||ch!="#") { // 当操作符栈为空且队列已经空了之后才结束循环
if(isp.find(ch)==isp.end()) { // 判断是否为数字，不是则直接加入后缀表达式
out.push(ch);
ch = in.front();
in.pop();
continue;
}
if(icp[ch] > isp[stk.top()]) { // 操作符优先级较低则直接加入栈中
stk.push(ch);
ch = in.front();
in.pop();
}else if(icp[ch] < isp[stk.top()]) { // 操作符优先级较高则弹出栈顶操作符加入后缀表达式
out.push(stk.top());
stk.pop();
}else { // 相等则弹出栈顶的左括号或右括号
stk.pop();
if(ch!="#") {
ch = in.front();
in.pop();
}
}
}
stack<int> sk;// 存储数字的栈
//TODO:逆波兰式子求解
while(!out.empty()) {// 遍历后缀表达式求值
string cur = out.front();
out.pop();
if(cur == "$") {// 处理单目运算符
int a = sk.top();
sk.pop();
sk.push(-a);
}else if(cur == "+") {
int a2 = sk.top();
sk.pop();
int a1 = sk.top();
sk.pop();
sk.push(a1+a2);
}else if(cur == "-") {
int a2 = sk.top();
sk.pop();
int a1 = sk.top();
sk.pop();
sk.push(a1-a2);
}else if(cur == "*") {
int a2 = sk.top();
sk.pop();
int a1 = sk.top();
sk.pop();
sk.push(a1*a2);
}else if(cur == "/") {
int a2 = sk.top();
sk.pop();
int a1 = sk.top();
sk.pop();
sk.push(a1/a2);
}else {
sk.push(stoi(cur));
}
}
return sk.top();
}
};
上述代码实现的是一个基本的四则运算计算器，其核心思路是将中缀表达式转化为后缀表达式，再根据后缀表达式求值得到结果。这个过程可以概括为：（GPT生成）
- 初始化两个栈，一个操作符栈stk和一个数字栈sk，初始化一个队列out用来存储后缀表达式；
- 遍历中缀表达式，对于每个字符进行如下判断：
- 如果是数字，直接入队out；
- 如果是左括号，入栈stk；
- 如果是右括号，弹出操作符栈中的元素并加入队列out，直到遇到左括号；注意：左右括号都不入队out；
- 如果是操作符，则判断其与操作符栈顶元素的优先级关系，如果优先级高，则压入操作符栈，否则将操作符栈中的元素弹出，加入队列out，然后继续比较栈顶元素和当前操作符的优先级，直至当前操作符可以入栈；
- 当遍历完中缀表达式后，如果操作符栈中还有元素，则将其全部依次弹出，并加入队列out；
- 此时out中存储的就是后缀表达式，对其进行求值即可得到结果。
例如，对于中缀表达式"3+42/(1-5)#"，按照上述算法可得到后缀表达式"34215-/+"。具体过程如下：
- 初始化操作符栈和队列out，将中缀表达式转化为一个字符数组arr；
- 从左到右遍历arr，对于每个元素进行判断：
- 当前元素是数字，直接加入out中；
- 当前元素是'('，压入操作符栈中；
- 当前元素是')'，弹出操作符栈中的元素并加入out，直至遇到'('；
- 当前元素是运算符，比较其与操作符栈顶元素的优先级，如果优先级高，则将其压入操作符栈；否则，依次弹出操作符栈中的元素并加入out，直至当前运算符可以入栈；
- 遍历完arr后，将操作符栈中的剩余元素全部弹出，并加入out；
- 对out中的元素进行求值，得到最终结果11。
在代码的 removeBlank 函数中，采用了一个简单的遍历字符串的方式来去除空格。具体来说，将原字符串中非空格的字符依次加入一个新的字符串中即可。
在处理单目运算符时，则需要考虑到单目运算符和减号（二元运算符）的区别。我们可以通过一个 bool 变量 pre 来判断当前字符是否为单目运算符。具体来说，如果 pre 为 true，且当前字符为减号，则说明其为单目运算符；否则，当前字符为减号则表示其为二元操作符。当遇到单目运算符时，我们可以将其替换为 "$"，在后面进行求值时再做特殊处理即可。
另外，在这个函数中还使用了一个辅助变量 push_src 来存储当前正在处理的数字串。当遇到运算符或者结束符号 # 时，就将该数字串加入队列 res 中。同时，pre 的取值根据当前字符是不是右括号进行相应的修改。
括号展开+栈
这个是LeetCode的官方解法
class Solution {
public:
int calculate(string s) {
stack<int> ops; // 用于存储括号内的符号
ops.push(1); // 先将符号入栈，初始为1
int sign = 1; // 当前数字的符号，默认为正数
int ret = 0; // 最终结果
int n = s.length(); // 字符串长度
int i = 0; // 当前遍历到字符串中的位置
while (i < n) {
if (s[i] == ' ') { // 空格直接跳过
i++;
} else if (s[i] == '+') { // 遇到加号，更新符号为栈顶的符号
sign = ops.top();
i++;
} else if (s[i] == '-') { // 遇到减号，更新符号为栈顶的符号的相反数
sign = -ops.top();
i++;
} else if (s[i] == '(') { // 遇到左括号，将当前符号入栈
ops.push(sign);
i++;
} else if (s[i] == ')') { // 遇到右括号，弹出栈顶符号
ops.pop();
i++;
} else { // 遇到数字，计算出完整的数字，并与当前符号相乘后累加到最终结果
long num = 0;
while (i < n && s[i] >= '0' && s[i] <= '9') {
num = num * 10 + s[i] - '0';
i++;
}
ret += sign * num;
}
}
return ret; // 返回最终结果
}
};
上述代码基于栈的思想实现了对带有括号的数学表达式的计算。其核心思路如下：
1.使用一个操作符栈 ops 存储括号内的符号，初始时将 1 入栈表示整个表达式的符号为正号。
2.遍历表达式中的每个字符，分别处理以下几种情况：
如果遇到空格，直接跳过。
如果遇到加号 +，则更新当前符号为栈顶的符号，并继续向后遍历。
如果遇到减号 -，则更新当前符号为栈顶的符号的相反数，并继续向后遍历。
如果遇到左括号 (，则将当前符号入栈，并继续向后遍历。
如果遇到右括号 )，则弹出栈顶符号，并继续向后遍历。
如果遇到数字，则计算出完整的数字，并与当前符号相乘后累加到最终结果，并继续向后遍历。
3.最终返回最终结果即可。
这样的实现可以处理带有括号的复杂表达式，同时也考虑到了符号的影响。
举一个例子，计算表达式 "1 + (2 - 3) - 4"
首先初始化操作符栈 ops，将符号 1 入栈：ops: [1]
然后从左到右遍历表达式中的每个字符，依次进行处理：
遇到数字 1，累加到最终结果 ret 中，此时 ret=1。
遇到空格，直接跳过。
遇到加号 +，更新当前符号为 ops 栈顶的符号 1，此时 sign=1，继续向后遍历。
遇到左括号 (，将当前符号 sign=1 入栈，此时 ops: [1, 1]，继续向后遍历。
遇到数字 2，由于此时栈顶符号为正号，所以累加到最终结果 ret 中，此时 ret=3。
遇到减号 -，更新当前符号为栈顶的符号的相反数 -1，此时 sign=-1，继续向后遍历。
遇到数字 3，由于此时栈顶符号为负号，所以将其乘以 (-1) 并累加到最终结果 ret 中，此时 ret=2。
遇到右括号 )，弹出栈顶符号，并继续向后遍历，此时 ops: [1]。
遇到减号 -，更新当前符号为栈顶的符号的相反数 -1，此时 sign=-1，继续向后遍历。
遇到数字 4，由于此时栈顶符号为负号，所以将其乘以 (-1) 并累加到最终结果 ret 中，此时 ret=-3。
最终返回结果 ret=-3
