Implement a basic calculator to evaluate a simple expression string.
The expression string contains only non-negative integers, +
, -
, *
, /
operators and empty spaces . The integer division should truncate toward zero.
You may assume that the given expression is always valid.
Some examples:
"3+2*2" = 7 " 3/2 " = 1 " 3+5 / 2 " = 5
Note: Do not use the eval
built-in library function.
class Solution {
public:
int getNextNum(vector<char> &str, int &idx) {
int leftPos, rightPos;
leftPos = rightPos = idx;
while (rightPos < str.size() && (str[rightPos] >='0' && str[rightPos] <= '9')) {
rightPos++;
}
int ret = 0;
for (int i=leftPos; i<rightPos; i++) {
ret = ret*10+(str[i]-'0');
}
idx = rightPos;
return ret;
}
int calculate(string s) {
vector<char> str;
for (int i=0; i<s.size(); i++)
if (s[i] != ' ')
str.push_back(s[i]);
int i=0;
stack<int> tmpResult;
int preNum, nextNum;
preNum = getNextNum(str, i);
tmpResult.push(preNum);
while (i<str.size()) {
int curIdx = i;
nextNum = getNextNum(str, ++i);
if (str[curIdx] == '*') {
tmpResult.pop();
tmpResult.push(preNum*nextNum);
}
else if (str[curIdx] == '/') {
tmpResult.pop();
tmpResult.push(preNum/nextNum);
}
else if (str[curIdx] == '-') {
tmpResult.push(-nextNum);
}
else {
tmpResult.push(nextNum);
}
preNum = tmpResult.top();
}
int retValue = tmpResult.top();
tmpResult.pop();
while (tmpResult.size() > 0) {
retValue += tmpResult.top();
tmpResult.pop();
}
return retValue;
}
};