Basic Calculator

发表于 | 分类于

Basic Calculator

题目描述：

具体描述可以看LeetCode227

解题思路：

代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
class Solution {
public:
    int calculate(string s) {
        stack<int> stk;
        char sign = '+';
        int tmp = 0;
        int res = 0;
        for (int i = 0; i < s.size(); i++){
            if (isdigit(s[i]))
                tmp = tmp * 10 + s[i] - '0';
            if (!isdigit(s[i]) && !isspace(s[i]) || i == s.size() - 1){
                if (sign == '+')
                    stk.push(tmp);
                else if (sign == '-')
                    stk.push(-tmp);
                else{
                    int num;
                    if (sign == '*')
                        num = stk.top() * tmp;
                    else
                        num = stk.top() / tmp;
                    stk.pop();
                    stk.push(num);
                }
                sign = s[i];
                tmp = 0;
            }
        }
        while(!stk.empty()){
            res += stk.top();
            stk.pop();
        }  
        return res;
        
    }
};