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

﹏ヽ暗。殇╰゛Y 2022-06-02 12:20 234阅读 0赞

\[编程|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;  
    }