-
Notifications
You must be signed in to change notification settings - Fork 0
/
BasicCalculator.cpp
81 lines (70 loc) · 2.5 KB
/
BasicCalculator.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
#include<iostream>
#include<string>
#include<vector>
using namespace std;
class Solution {
public:
int calculate(string s) {
vector<int> stack;
vector<int> operat;
int num = 0;
char lastOperator = 'x';
for(int i=0;i<s.length();i++){
cout<<"numStack: ";
for(int j=0;j<stack.size();j++) cout<<stack[j]<<" ";
cout<<endl;
cout<<"operator: ";
for(int j=0;j<operat.size();j++) cout<<operat[j]<<" ";
cout<<endl;
cout<<"num: "<<num<<endl;
if(s[i] == ' ') continue;
if(s[i] >= '0' && s[i] <= '9'){
num = num * 10 + s[i] - '0';
}else if(s[i] == '+' || s[i] == '-'){
if(lastOperator == '(' || lastOperator == 'x'){
s[i] == '+' ? operat.push_back(1) : operat.push_back(-1);
stack.push_back(num);
}else if(lastOperator == '+' || lastOperator == '-'){
num = num * operat.back() + stack.back();
operat.pop_back();
stack.pop_back();
stack.push_back(num);
s[i] == '+' ? operat.push_back(1) : operat.push_back(-1);
}else if(lastOperator == ')'){
if(!operat.empty()){
num = num * operat.back() + stack.back();
operat.pop_back();
stack.pop_back();
}
stack.push_back(num);
s[i] == '+' ? operat.push_back(1) : operat.push_back(-1);
}
num = 0;
lastOperator = s[i];
}else if(s[i] == '('){
num = 0;
lastOperator = s[i];
}else if(s[i] == ')'){
if(!operat.empty() && lastOperator != '('){
num = num * operat.back() + stack.back();
operat.pop_back();
stack.pop_back();
//stack.push_back(num);
}
lastOperator = s[i];
}
}
cout<<"num: "<<num<<endl;
if(stack.empty()) return num;
if(operat.empty()) return stack.back();
else return stack.back() + num * operat.back();
}
};
int main(){
string s;
getline(cin, s);
//cout<<s;
Solution *solution = new Solution();
cout<<solution->calculate(s);
return 0;
}