一、思路
给出一个后缀表达式(以字符串数组的形式,每一项要么是一个操作数字符串,要么是一个运算符),计算过程中要将操作数从string形式转换为int形式,在计算表达式运算结果时,需要设置一个栈 num 保存操作数,计算步骤如下:
1)若当前项是操作数,则将该操作数压入num栈中;
2)若当前项是运算符,则从操作数栈中出栈两个操作数(栈顶的数作为第二个操作数,次栈顶的数作为第一个操作数),计算两个操作数进行运算的结果,然后将结果压入操作数栈num中。
遍历完后缀表达式数组后,操作数栈num中必定只剩下一个数(表达式正确的情况下),这个数就是表达式的运算结果。
二、代码
int solve(vector<string> postexp;) {
//postexp是后缀表达式,每一项是一个操作数串或者运算符
stack<int> num; //操作数栈
int len = postexp.size();
int i = 0;
while(i < len){ //对后缀表达式求值
if(postexp[i] != "+" && postexp[i] != "-" && postexp[i] != "*"){
int tmp = atoi(postexp[i].c_str()) ; //操作数(string类型转换为int型)
num.push(tmp); //将操作数压入栈中
i++;
}
else{ //每遇到一个操作符,就将两个操作数出栈,运算完后将结果压栈
int opnum1 = num.top();
num.pop();
int opnum2 = num.top();
num.pop();
int res = 0; //运算结果
char op= postexp[i][0]; //字符串的第一个字符(实际上也只有一个字符)
switch(op){ //注意两个操作数的顺序,栈顶元素作为第二个操作数
case '+':
res = opnum2 + opnum1;
break;
case '-':
res = opnum2 - opnum1;
break;
case '*':
res = opnum2 * opnum1;
break;
case '/':
res = opnum2 / opnum1;
break;
default:
break;
}
num.push(res);
i++;
}
}
return num.top(); //最后栈中必定只剩下一个数,就是运算结果
}