华为机试：四则运算

一、题目：

描述

输入一个表达式（用字符串表示），求这个表达式的值。

保证字符串中的有效字符包括[‘0’-‘9’],‘+’,‘-’, ‘*’,‘/’ ,‘(’， ‘)’,‘[’, ‘]’,‘{’ ,‘}’。且表达式一定合法。

输入描述：

输入一个算术表达式

输出描述：

得到计算结果

二、输入输出示例：

三、思路：

用栈，将中缀表达式转为后缀表达式，然后在计算后缀表达式的值。

四、代码：

#include<iostream>
using namespace std;
#include<stack>
#include<string>
#include<vector>
#include<cmath>
int main()
{
	string s;
	while (cin >> s)
	{
		//将中缀表达式转换为后缀表达式
		string s1;     //后缀表达式
		vector<int> v; //记录参与运算数字的位数
		stack<char> op; //记录运算符号
		for (int i = 0; i < s.size();)
		{
			int count = 0;
			if (s[i] >= '0' && s[i] <= '9')
			{
				while (s[i] >= '0' && s[i] <= '9')
				{
					s1 += s[i];
					count++;
					i++;
				}
				v.push_back(count);
			}
			else
			{
				if (s[i] == '+' || s[i] == '-')
				{
					if (s[i] == '-' && (s[i - 1] == '(' || s[i - 1] == '[' || s[i - 1] == '{')) //减号前面是括号，则需要补0，为了不影响位数的记录，补字母o
					{
						s1 += 'o';
					}
					while (!op.empty() && op.top() != '(' && op.top() != '[' && op.top() != '{')//加减优先级最低，堆栈不空并且栈顶没有括号，则输出栈顶元素
					{
						s1 += op.top();
						op.pop();
					}
					op.push(s[i]);
				}
				else if (s[i] == '*' || s[i] == '/')
				{
					while (!op.empty() && (op.top() == '*' || op.top() == '/'))   //乘除的优先级高，如果栈顶元素是乘除，则输出到不是乘除为止
					{
						s1 += op.top();
						op.pop();
					}
					op.push(s[i]);
				}
				else if(s[i] == '(' || s[i] == '[' || s[i] == '{')   //如果是前括号，则直接存入堆栈
				{
					op.push(s[i]);
				}
				//遇到后括号要直接输出符号到遇到前括号为止
				else if (s[i] == ')')                                
				{
					while (op.top() != '(')
					{
						s1 += op.top();
						op.pop();
					}
					op.pop();
				}
				else if (s[i] == ']')
				{
					while (op.top() != '[')
					{
						s1 += op.top();
						op.pop();
					}
					op.pop();
				}
				else if (s[i] == '}')
				{
					while (op.top() != '{')
					{
						s1 += op.top();
						op.pop();
					}
					op.pop();
				}
				i++;
			}
		}
		while (!op.empty())
		{
			s1 += op.top();
			op.pop();
		}
		//计算后缀表达式
		int flag = 0;      //flag表示后缀表达式字符串的下标
		int flag1 = 0;     //flag1表示记录数的位数的vector容器的下标
		stack<int> res;    //存储后缀表达式中参与运算的数
		while (flag < s1.size())
		{
			int sum = 0;   //计算位数不是1的数
			if (s1[flag] >= '0' && s1[flag] <= '9')
			{
				for (int i = 0; i < v[flag1]; i++)
				{
					sum += (s1[flag] - '0') * pow(10, v[flag1] - 1 - i);
					flag++;
				}
				res.push(sum);
				flag1++;
			}
			else if (s1[flag] == 'o')
			{
				res.push(0);
				flag++;
			}
			else
			{
				int b = res.top();   //b为后缀表达式中运算符号前面的第1个数     
				res.pop();
				int a = res.top();   //a为后缀表达式中运算符号前面的第2个数
				res.pop();                
				switch (s1[flag])
				{
					case '+':
					{
						res.push(a+b);
						break;
					}
					case '-':
					{
						res.push(a-b);
						break;
					}
					case '*':
					{
						res.push(a*b);
						break;
					}
					case '/':
					{
						res.push(a/b);
						break;
					}
				}
				flag++;
			}
		}
		cout << res.top() << endl;
	}
	system("pause");
	return 0;
}