[编程|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,然后判断。
代码如下
[cpp] view plain copy
#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;
}