牛客 BM49 表达式求值

最新推荐文章于 2024-10-19 11:17:13 发布

JehanRio

最新推荐文章于 2024-10-19 11:17:13 发布

阅读量399

点赞数 1

分类专栏： Leetcode专栏文章标签：算法职场和发展

本文链接：https://blog.csdn.net/weixin_51322383/article/details/132425630

版权

Leetcode专栏专栏收录该内容

278 篇文章 2 订阅

订阅专栏

文章目录

题目
代码（8.22 首刷看解析）

题目

在这里插入图片描述
牛客 BM49 表达式求值

代码（8.22 首刷看解析）

class Solution {
public:
    int solve(string s) {
        stack<int> sk;
        stack<char> op;
        int i = 0;
        while(i < s.size()) {
            if(isdigit(s[i])) {
                int num = 0;
                while(isdigit(s[i])) {
                    num = num*10 + s[i]-'0';
                    i++;
                }
                sk.push(num);
            } else if(s[i] == '(') {
                op.push(s[i]);
                i++;
            } else if(s[i] == ')') {
                while(op.top() != '(') {
                    int num1 = sk.top();
                    sk.pop();
                    int num2 = sk.top();
                    sk.pop();
                    char cal = op.top();
                    op.pop();
                    sk.push(calculate(num1, num2, cal));
                }
                op.pop();
                i++;
            } else if(s[i] == '*') {
                op.push(s[i]);
                i++;
            } else {    // + -
                while(sk.size() >= 2) {
                    if(op.top() == '(')    // 不能越过小括号
                        break;
                    int num1 = sk.top();
                    sk.pop();
                    int num2 = sk.top();
                    sk.pop();
                    char cal = op.top();
                    op.pop();
                    sk.push(calculate(num1, num2, cal));
                }
                op.push(s[i]);
                i++;
            }   
        }
        while(!op.empty()) {
            int num1 = sk.top();
            sk.pop();
            int num2 = sk.top();
            sk.pop();
            char cal = op.top();
            op.pop();
            sk.push(calculate(num1, num2, cal));
        }
        return sk.top();
    }
    int calculate(int num1, int num2, char cal) {
        if(cal == '+')
            return num1+num2;
        if(cal == '-')
            return num2 - num1;
        return num1*num2;
    }
};