【leetcode】227 基本计算器 II（栈）

题目链接：https://leetcode-cn.com/problems/basic-calculator-ii/comments/

题目描述

实现一个基本的计算器来计算一个简单的字符串表达式的值。

字符串表达式仅包含非负整数，+， - ，*，/ 四种运算符和空格 。 整数除法仅保留整数部分。

示例 1:

输入: "3+2*2"
输出: 7

示例 2:

输入: " 3/2 "
输出: 1

示例 3:

输入: " 3+5 / 2 "
输出: 5

说明：

• 你可以假设所给定的表达式都是有效的。
• 请不要使用内置的库函数 eval。

思路

一般做法需要符号栈、数据栈两个栈；在该题目中可以只使用一个数据栈，符号栈用一个变量sign来存储上一个符号。主要思想如下：
（1）输入的数字可能>10
（2）将减法转化为加法（取相反数）
（3）乘除法的优先级比加减法高，直接计算

注意：
（1）加减乘除空格的ASCII码都小于’0’，ASCII对照表如下：http://tool.oschina.net/commons?type=4
（2）先做减法，避免int溢出
（3）表达式中没有括号
（4）char类型，不能使用switch

代码

class Solution {
public:
    int calculate(string s) {
        int num = 0;
        char sign = '+';
        stack<int> nums;

        for (int i = 0; i < s.size(); ++i) {
            // 数字位；加减乘除和空格ASCII码都小于'0'
            if(s[i] >= '0'){
                num = num * 10   -'0' + s[i]; // 先做减法防止溢出
            }
            // 符号位或最后一字符（已得到最后的数字）
            if((s[i] < '0' && s[i] != ' ') || i == s.size() -1){
                if(sign == '+')
                    nums.push(num);
                else if(sign == '-')
                    nums.push(-num);
                else if(sign == '*'||sign == '/'){
                    int tmp = sign == '*'? nums.top()*num : nums.top()/num;
                    nums.pop();
                    nums.push(tmp);
                }
                sign = s[i];    //保存当前符号
                num = 0;
            }
        }

        int ret=0;  // 求和计算最终结果
        while (!nums.empty()){
            ret += nums.top();
            nums.pop();
        }
        return ret;
    }
};