题目
实现一个基本的计算器来计算一个简单的字符串表达式的值。
字符串表达式仅包含非负整数,+, - ,*,/ 四种运算符和空格 。 整数除法仅保留整数部分。
示例 1:
输入: “3+2*2”
输出: 7
示例 2:
输入: " 3/2 "
输出: 1
示例 3:
输入: " 3+5 / 2 "
输出: 5
说明:
你可以假设所给定的表达式都是有效的。
请不要使用内置的库函数 eval。
解题思路:
设置一个栈,从左到右遍历s中的一段一段的值,如果是前面是正号,则直接push进去,如果是负号,则push它的相反数,如果是乘号,则让这个数字乘以栈顶元素,如果是除号,则让栈顶元素除以这个数,最后把栈里面的元素都加起来,就得到了最终的结果。
在代码中,需要设置一个start来索引开始找新的数字段的开头和数字段开头的对应的负号。
class Solution
{
public:
int calculate(string s)
{
int res = 0;
stack<int> my_stack{};
vector<char> my_vec{};
for (auto x : s)
{
if (x != ' ') my_vec.push_back(x);
}
int start = 0;
int end = 0;
int tmp = 0;
while(start != my_vec.size())
{
tmp = 0;
for (int i = start; i < my_vec.size(); i++)
{
if (my_vec[i] >= '0')
{
if (i == 0)
{
tmp = my_vec[i] - '0';
}
else if (my_vec[i - 1] == '+' || my_vec[i - 1] == '*' || my_vec[i - 1] == '/') //前面是这三个符号,新的tmp都应该为正
tmp = my_vec[i] - '0';
else if (my_vec[i - 1] == '-')
tmp = -1 * (my_vec[i] - '0');
else if (my_vec[i - 1] >= '0')
{
if (tmp >= 0)
tmp = tmp * 10 + (my_vec[i] - '0');
else
tmp = tmp * 10 - (my_vec[i] - '0');
}
if (i == my_vec.size() - 1)
end = i;
}
else
{
end = i;
break;
}
}
if (start == 0)
my_stack.push(tmp);
else
{
if (my_vec[start - 1] == '+' || my_vec[start - 1] == '-')
my_stack.push(tmp);
else if (my_vec[start - 1] == '*')
my_stack.top() = my_stack.top() * tmp;
else if (my_vec[start - 1] == '/')
my_stack.top() = my_stack.top() / tmp;
}
start = end + 1;
}
while (!my_stack.empty())
{
res += my_stack.top();
my_stack.pop();
}
return res;
}
};