题解 | #表达式求值#

#include <iostream>
#include <string>
#include <stack>
using namespace std;

int calculate(int a, int b, char sym){
    switch(sym){
        case '+':
            return a + b;
            break;
        case '-':
            return a - b;
            break;
        case '*':
            return a * b;
            break;
        case '/':
            return a / b;
            break;
        default:
            return 0;
            break;
    }
}

int process(string s){
    int flag = 0; //0无符号，1为正号，2为负号
    stack<int> numst;
    stack<char> symbolst;
    
    for(int i = 0; i < s.size(); i++){
        if(isdigit(s[i])){
            int j = i; int num = 0;
            while(i + 1 < s.size() && isdigit(s[i + 1])) i++;
            string tmp = s.substr(j, i - j + 1);
            for(int k = 0; k < tmp.size(); k++){
                num = num * 10 + (tmp[k] - '0');
            }   
            
            if(flag == 2) num = 0 - num;
            flag = 0;
            numst.push(num);
        }
        else if(s[i] == '*' || s[i] == '/' || s[i] == '('){
            symbolst.push(s[i]);
        }
        else if(s[i] == '+' || s[i] == '-'){
            if(i == 0 || s[i - 1] == '('){
                if(s[i] == '+') flag = 1;
                else flag = 2;
            }
            
            while(!flag && !symbolst.empty() && symbolst.top() != '('){
                int b = 0, a = 0;
                char sym_tmp;
                b = numst.top(); numst.pop();
                a = numst.top(); numst.pop();
                sym_tmp = symbolst.top(); symbolst.pop();
                numst.push(calculate(a, b, sym_tmp));
            }
            
            if(!flag) symbolst.push(s[i]); //
        }
        else if(s[i] == ')'){
            while(symbolst.top() != '('){
                int b = 0, a = 0;
                char sym_tmp;
                b = numst.top(); numst.pop();
                a = numst.top(); numst.pop();
                sym_tmp = symbolst.top(); symbolst.pop();
                numst.push(calculate(a, b, sym_tmp));
            }
            symbolst.pop();
        }
        else{
            cout << "error!" << endl;
        }
    }
    
    while(!symbolst.empty()){
        int b = 0, a = 0;
        char sym_tmp;
        b = numst.top(); numst.pop();
        a = numst.top(); numst.pop();
        sym_tmp = symbolst.top(); symbolst.pop();
        numst.push(calculate(a, b, sym_tmp));
    }
    
    return numst.top();
}

int main(){
    string str = "";
    while(cin >> str){
        int res = process(str);
        cout << res << endl;
    }
    
    
    return 0;
}