Codeforces 552E - Vanya and Brackets (中缀表达式求值)

左括号一定在*和+之间,右括号一定在+和+乘之间。乘号最多15个所以可以暴力枚举括号位置。


#include <iostream>
#include <cstdio>
#include <cstring>
#include <stack>
#define LL long long
using namespace std;
#define maxn 5005
char s[maxn];

int ps1[maxn];
int ps2[maxn];

LL num[maxn];
char sig[maxn];
int nn,sn;
int len;
int pri(char c){
    if(c=='+'||c=='-') return 1;
    if(c=='*'||c=='/') return 2;
    if(c=='(') return 0;
}

void cal(){
    if(sig[sn]=='+') num[nn-1]=num[nn]+num[nn-1];
    if(sig[sn]=='-') num[nn-1]=num[nn-1]-num[nn];
    if(sig[sn]=='*') num[nn-1]=num[nn-1]*num[nn];
    if(sig[sn]=='/') num[nn-1]=num[nn-1]/num[nn];
    nn--;
    sn--;
}


LL solve(int l,int r){
    if(r<l) return 0;
    nn=sn=0;
    int curn=0;
    for(int i=0;i<=len;i++){
        if(i==l) sig[++sn]='(';
        if(isdigit(s[i])){
            curn=curn*10+s[i]-'0';
        }
        else {
            if(i-1!=r){
                num[++nn]=curn;
                curn=0;
            }
            if(sn&&pri(s[i])<=pri(sig[sn])) cal();
            sig[++sn]=s[i];
        }
        if(i==r) {
            num[++nn]=curn;
            curn=0;
            while(sig[sn]!='(') cal();
            sn--;
        }

    }
    if(len!=r) num[++nn]=curn;
    while(nn>1) cal();
    return num[1];
}

int main(){
    while(~scanf("%s",s)){
        int n1=0,n2=0;
        len=strlen(s);
        LL res=solve(0,len-1);
        ps1[++n1]=0;
        for(int i=1;i<len-2;i+=2){
            if(s[i]=='*'&&s[i+2]=='+') ps1[++n1]=i+1;
        }
        for(int i=1;i<len-2;i+=2){
            if(s[i]=='+'&&s[i+2]=='*') ps2[++n2]=i+1;
        }
        ps2[++n2]=len-1;
        int clen;
        for(int i=1;i<=n1;i++){
            for(int j=1;j<=n2;j++){
                int s=ps1[i];
                int t=ps2[j];
                if(s>t) continue;
                res=max(res,solve(s,t));
            }
        }
        printf("%I64d\n",res);
    }
    return 0;
}


你可能感兴趣的:(中缀表达式)