2017华为机试（仿LISP字符串运算）

[编程|300分] 仿LISP字符串运算 
题目描述 
LISP语言唯一的语法就是括号要配对。 
形如 (OP P1 P2 …)，括号内元素由单个空格分割。 
其中第一个元素OP为操作符，后续元素均为其参数，参数个数取决于操作符类型 
注意：参数 P1, P2 也有可能是另外一个嵌套的 (OP P1 P2 …) 
当前OP类型为add/sub/mul/div(全小写)，分别代表整数的加减乘除法。简单起见，所以OP参数个数为2
举例
-输入：(mul 3 -7)输出：-21
输入：(add 1 2) 输出：3
输入：(sub (mul 2 4) (div 9 3)) 输出 ：5
输入：(div 1 0) 输出：error
常规方法是用两个栈分别存放操作数和操作符，本文用一个栈来实现，首先从后往前提取操作数和操作符存放在vector，然后判断。

代码如下

#include<iostream>#include <stack>#include<string>#include<vector>#include <algorithm>using namespace std;int main(){vector<string> st;stack<int> res;string str = "(sub (mul 2 4) (div 9 3))";//getline(cin, str);bool flag = true;int N = str.size();for (int i = N - 1; i >=0;){if (str[i] == ')' || str[i] == ' '||str[i] == '(')i=i-1;else if (str[i] >= 'a'&&str[i] <= 'z'){string op = str.substr(i - 2, 3);st.push_back(op);i -= 3;}else{int spaceIndex = str.rfind(' ', i);if ((spaceIndex<0)||(str[spaceIndex+1]<='z'&&str[spaceIndex+1] >= 'a'))spaceIndex = str.rfind('(', i);string num = str.substr(spaceIndex + 1, i - spaceIndex);st.push_back(num);i = spaceIndex;}}int n = st.size();for (int j = 0; j<n; ++j){string s = st[j];if (s == "add" || s == "sub" || s == "mul" || s == "div"){if (res.size()<2)return 0;int num2 = res.top(); res.pop();int num1 = res.top(); res.pop();int result = 0;if (s == "add")result = num1 + num2;else if (s == "sub")result = num2 - num1;else if (s == "mul")result = num1 * num2;else if (s == "div"){if (num1 == 0)flag = false;elseresult = num2 / num1;}res.push(result);}else{res.push(atoi(s.c_str()));}}if (!flag)cout << "error" << endl;elsecout<<res.top()<<endl;}