C++表达式求值（栈）

AcWing第3302题

题目

给定一个表达式，其中运算符仅包含 +,-,*,/（加 减 乘 整除），可能包含括号，请你求出表达式的最终值。

注意：

• 数据保证给定的表达式合法。
• 题目保证符号 - 只作为减号出现，不会作为负号出现，例如，-1+2,(2+2)*(-(1+1)+2) 之类表达式均不会出现。
• 题目保证表达式中所有数字均为正整数。
• 题目保证表达式在中间计算过程以及结果中，均不超过 231−1。
• 题目中的整除是指向 0 取整，也就是说对于大于 0 的结果向下取整，例如 5/3=1，对于小于 0 的结果向上取整，例如 5/(1−4)=−1。
• C++和Java中的整除默认是向零取整；Python中的整除//默认向下取整，因此Python的eval()函数中的整除也是向下取整，在本题中不能直接使用。

输入格式

共一行，为给定表达式。

输出格式

共一行，为表达式的结果。

数据范围

表达式的长度不超过 105。

输入样例：

(2+2)*(1+1)

输出样例：

8

解题思路

首先，分析题意，首先是考虑到表达式中有两种元素，一种为运算的数据，一种为运算符，其中运算符需要考虑优先级。所以采用双栈stack nums;
stack op;来存放表达式的各个数据，采用哈希表存放运算符的优先级。
1. 将表达式的数据以及运算符分别入栈。

• 运算数直接入nums栈，要注意的是多位数的扫描判定。
• 左括号可以直接入栈，右括号应应扫描op栈，如果在左括号前存在运算符，应当先计算改元素符）右括号不用入栈。
• 剩下的运算符“ + - * / ”入栈。入栈前先判定op栈内运算符是否存在以及优先级是否不低于当前运算符，如果栈定运算符存在且优先权不低于准备入栈元素应当计算栈定运算（遵循了从左至右的优先级）。

2. 将op栈还存在的运算都取出来计算。

3. eval()运算。

• 从nums栈中取出两个数出栈（注意：先出来的数为第二位运算数）。
• 从op栈中取出运算符。
• 判定运算符类型并计算。
• 将运算结果入nums栈。

实验代码

#include <iostream>
#include <cstring>
#include <algorithm>
#include <stack>
#include <unordered_map>

using namespace std;

stack<int> nums; //运算数的栈
stack<char> op;  // 运算符的栈

void eval(){
    auto a = nums.top();nums.pop(); //第二个运算数
    auto b = nums.top();nums.pop(); //第一个运算数
    auto c = op.top();op.pop(); //运算符
    int res; //当前的运算结果
    if(c == '+') res = a + b;
    if(c == '-') res = b - a;
    if(c == '*') res = a * b;
    if(c == '/') res = b / a;
    nums.push(res); //此次运算结果入栈
}
int main(){
    unordered_map<char,int> priority = {{'+',1},{'-',1},{'*',2},{'/',2}}; //存放运算符的优先级

    string str; //整个表达式
    cin >> str;
    // int len = str.size(); //表达式长度，字符个数
    for(int i = 0; i < str.size(); i++){
        //逐位遍历整个表达式，将数据入栈
        auto c = str[i];
        if(isdigit(c)){ // 判断c是否为数字
            int x = 0, j = i;
            while(j < str.size() && isdigit(str[j])) //找出数字后一位是否也为数字
                x = x*10 + str[j ++] - '0'; //将运算数据char类型转换为int
            i = j - 1; //更新最先的位置
            nums.push(x); //运算数入栈
        }
        else if(c == '(') op.push(c); //左括号直接入栈
        else if(c == ')'){
            while(op.top() != '(') eval(); //括号内运算先算出结果
            op.pop(); //运算符栈顶元素出栈
        }
        else{ //其他运算符 + - * /
            //运算符c前面还有运算符，且不是'('并且运算优先级比不低于c，则直接计算运算符栈定元素
            while(op.size() && op.top() != '(' && priority[op.top()] >= priority[c]) eval();
            op.push(c); //当前运算符入栈
        }
    }
    while(op.size()) eval();
    cout << nums.top() << endl;
    return 0;
}

总结

有很多的小细节，对读题能力以及栈的理解使用有很大帮助。