题解 | #四则运算#

#include<bits/stdc++.h>
using namespace std;
stack<int>nums;
stack<char>ops;
map<char,int>pri={{'+',1},{'-',1},{'*',2},{'/',2}};

void eval(){
    int b=nums.top();
    nums.pop();
    int a=nums.top();
    nums.pop();
    char op=ops.top();
    ops.pop(); 
    if(op=='+')nums.push(a+b);
    else if(op=='-')nums.push(a-b);
    else if(op=='*')nums.push(a*b);
    else nums.push(a/b);
}

int main() {    
    bool isminus=true;
    string str;
    cin>>str;
    for(int i=0;i<str.size();i++){
        char ch=str[i];
        if(ch=='('||ch=='['||ch=='{')ch='(';
        if(ch==')'||ch==']'||ch=='}')ch=')';
        if(isdigit(ch)||(ch=='-'&&isminus)){
            int sign=1,j=i,num=0;
            if(!isdigit(ch))i++,sign=-1;
            while(j<str.size()&&isdigit(str[j]))num=10*num+str[j]-'0',j++;
            nums.push(num*sign),i=j-1,isminus=false;
        }
        else if(ch=='(')ops.push(ch),isminus=true;
        else if(ch==')'){
            while(ops.size()&&ops.top()!='(')eval();
            if(ops.size())ops.pop();
        }
        else {
            while(ops.size()&&(ops.top()!='(')&&pri[ops.top()]>=pri[ch])eval();
            ops.push(ch);
        }
    }
    while(ops.size())eval();
    cout<<nums.top();
    return 0;
}