-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbasic_calculator_iii.cpp
85 lines (70 loc) · 2.31 KB
/
basic_calculator_iii.cpp
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
class Solution {
public:
ll cal(stack<ll> & s, char op) {
ll n1 = 0;
ll n2 = 0;
if(s.size()) n1 = s.top(), s.pop();
if(s.size()) n2 = s.top(), s.pop();
ll ret = 0;
switch(op) {
case '+': ret = n1 + n2; break;
case '*': ret = n1 * n2; break;
case '-': ret = n1 - n2; break;
case '/': ret = n1 / n2; break;
}
return ret;
}
int calculate(string s) {
stack<ll> num;
stack<char> op;
string prev = "";
s = "("+s+")";
bool isLastWasOp = false;
for(int i = s.size() - 1; i >= 0; --i) {
if(s[i] == ' ') continue;
if('0' <= s[i] && s[i] <= '9') {
prev = string(1, s[i]) + prev;
isLastWasOp = false;
} else if(prev != "") {
num.push(stol(prev.c_str()));
prev = "";
}
if(s[i] == '(') {
stack<ll> tmpNum;
stack<char> tmpOp;
if(isLastWasOp) num.push(0);
while(op.top() != ')') {
if(op.top() == '*' || op.top() == '/') {
num.push(cal(num, op.top()));
op.pop();
}
else {
tmpNum.push(num.top());
num.pop();
tmpOp.push(op.top());
op.pop();
}
}
while(tmpOp.size()) {
num.push(tmpNum.top()), tmpNum.pop();
op.push(tmpOp.top()), tmpOp.pop();
}
while(op.top() != ')') {
num.push(cal(num, op.top()));
op.pop();
}
op.pop();
}
if(s[i] == ')' || s[i] == '-' || s[i] == '*' || s[i] == '+' || s[i] == '/') {
op.push(s[i]);
isLastWasOp = (s[i] == '-' || s[i] == '+');
}
}
return (int)num.top();
}
};
int main() {
}