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;
else
result = num2 / num1;
}

res.push(result);
}
else
{
res.push(atoi(s.c_str()));
}
}
if (!flag)
cout << "error" << endl;
else
cout<<res.top()<<endl;
}