leetcode 772.Basic Calculator III

最新推荐文章于 2024-12-12 22:36:07 发布

weixin_30835923

最新推荐文章于 2024-12-12 22:36:07 发布

阅读量245

点赞数

CC 4.0 BY-SA版权

文章标签： java

原文链接：http://www.cnblogs.com/newnoobbird/p/9627496.html

本文介绍了一种基于栈的数据结构实现表达式求值的方法，包括处理括号内表达式和四则运算，提供了两种不同的实现方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这道题就可以结合Basic Calculator中的两种做法了，分别是括号运算和四则运算的，则使用stack作为保持的结果，而使用递归来处理括号内的值的。

class Solution {
public:
    int calculate(string s) {
        int num=0,res=0,n=s.size();
        stack<int> st;
        char op='+';
        for(int i=0;i<n;i++){
            char c=s[i];
            if(c>='0' && c<='9'){
                num=num*10+c-'0';
            }else if(c=='('){
                int j=i,cnt=0;
                for(;i<n;i++){
                    if(s[i]=='(') cnt++;
                    if(s[i]==')') cnt--;
                    if(cnt==0) break;
                }
                num=calculate(s.substr(j+1,i-j-1));
            }
            if(s[i]=='+' || s[i]=='-' || s[i]=='*' || s[i]=='/' || i==n-1){
                if(op=='+') st.push(num);
                if(op=='-') st.push(-num);
                if(op=='*' || op=='/'){
                    int temp=(op=='*')?(st.top()*num):(st.top()/num);
                    st.pop();
                    st.push(temp);
                }
                op=s[i];
                num=0;
            }
        }
        while(!st.empty()){
            res+=st.top();
            st.pop();
        }
        return res;
    }
};

当然这个地方也是可以进行一个变化的，不使用栈，而是考虑用一个变量来保存上一个值的，即栈顶的值，然后继续进行处理的。

class Solution {
public:
    int calculate(string s) {
        int res=0,num=0,last=0,n=s.size();
        char op='+';
        for(int i=0;i<n;i++){
            char c=s[i];
            if(c>='0'&&c<='9'){
                num=num*10+c-'0';
            }
            else if(c=='('){
                int j=i,cnt=0;
                for(;i<n;i++){
                    if(s[i]=='(') cnt++;
                    if(s[i]==')') cnt--;
                    if(cnt==0) break;
                }
                num=calculate(s.substr(j+1,i-j-1));
            }
            if(c=='+' || c=='-' ||c=='*' ||c=='/' ||i==n-1){
                if(op=='+') last+=num;
                if(op=='-') last-=num;
                if(op=='*') last*=num;
                if(op=='/') last/=num;
                if(c=='+'||c=='-'||i==n-1){
                    res+=last;
                    last=0;
                }
                op=c;
                num=0;
            }
        }
        return res;
    }
};