[华为机试题]四则运算

题目描述

请实现如下接口

     /*  功能：四则运算

     *  输入： strExpression ：字符串格式的算术表达式，如 : "3+2*{1+2*[-4/(8-6)+7]}"

          *  返回： 算术表达式的计算结果

     */

     public   static   int calculate(String strExpression)

    {

         /*  请实现 */

         return  0;

    }  

约束：

1. pucExpression 字符串中的有效字符包括 [‘0’-‘9’],‘+’,‘-’, ‘*’,‘/’ ,‘(’， ‘)’,‘[’, ‘]’,‘{’ ,‘}’。

2. pucExpression 算术表达式的有效性由调用者保证; 

 

输入描述:

输入一个算术表达式

输出描述:

得到计算结果

输入例子:

3+2*{1+2*[-4/(8-6)+7]}

输出例子:

25

思路很明确：中缀表达式转后缀表达式，再计算后缀表达式的值即可，注意的是对于一元运算符‘-’的处理，和数字的位数要做一个记录

写了3个小时。。。。中间很多问题，调试了半天

代码;

#include<iostream>
#include<string>
#include<vector>
#include<stack>
using namespace std;
int main() {
	string s;
	while (cin >> s) {
		stack<char> opera;
		vector<int> numcnt;//用来保存每个数字的位数，以保证计算后缀表达式时的正确性
		string s1;//后缀表达式
	    //中缀表达式转后缀表达式
		for (int i = 0;i<s.size();i++) {
			if (s[i] >= '0'&&s[i] <= '9') {
				int tmp = 0;
				while (s[i] >= '0'&&s[i] <= '9') {
					tmp++;
					s1 += s[i];
					i++;
				}
				i--;
				numcnt.push_back(tmp);
			}
			else if (s[i] == '-' || s[i] == '+') {
				if (s[i] == '-' && (s[i - 1] == '(' || s[i - 1] == '[' || s[i - 1] == '{'))
					s1 += '0';
				while (!opera.empty()&&(opera.top() == '*' || opera.top() == '/' || opera.top() == '+' || opera.top() == '-')) {
					s1 += opera.top();
					opera.pop();
				}
				opera.push(s[i]);
			}
			else if (s[i] == '*' || s[i] == '/') {
				while (!opera.empty()&&(opera.top() == '*' || opera.top() == '/')) {
					s1 += opera.top();
					opera.pop();
				}
				opera.push(s[i]);
			}
			else if (s[i] == '(' || s[i] == '[' || s[i] == '{')
				opera.push(s[i]);
			else if (s[i] == ')') {
				while (opera.top() != '(') {
					s1 += opera.top();
					opera.pop();
				}
				opera.pop();
			}
			else if (s[i] == ']') {
				while (opera.top() != '[') {
					s1 += opera.top();
					opera.pop();
				}
				opera.pop();
			}
			else if (s[i] == '}') {
				while (opera.top() != '{') {
					s1 += opera.top();
					opera.pop();
				}
				opera.pop();
			}
			else
				cout << "Invalid input!" << endl;
		}
		while (!opera.empty()) {
			s1 += opera.top();
			opera.pop();
		}
		//计算后缀表达式的值
		stack<int> nums;
		int ind = 0;
		for (int i = 0;i<s1.size();i++) {
			if (s1[i] >= '0'&&s1[i] <= '9') {
				int total = 0;
				while (numcnt[ind]--)
					total = 10 * total + (s1[i++] - '0');
				i--;
				nums.push(total);
				ind++;
			}			
			else {
				int tmp1 = nums.top();
				nums.pop();
				int tmp2 = nums.top();
				nums.pop();
				if (s1[i] == '+')
					nums.push(tmp2 + tmp1);
				else if (s1[i] == '-')
					nums.push(tmp2 - tmp1);
				else if (s1[i] == '*')
					nums.push(tmp2*tmp1);
				else
					nums.push(tmp2 / tmp1);
			}
		}
		cout << nums.top() << endl;		
	}
}