LeetCode 227. 基本计算器 II--栈模拟

最新推荐文章于 2023-11-01 01:01:46 发布

Guapifang

最新推荐文章于 2023-11-01 01:01:46 发布

阅读量162

点赞数

分类专栏： LeetCode 文章标签：字符串 leetcode 栈

本文链接：https://blog.csdn.net/weixin_43918046/article/details/114109555

版权

LeetCode 专栏收录该内容

481 篇文章 5 订阅

订阅专栏

基本计算器 II

实现一个基本的计算器来计算一个简单的字符串表达式的值。

字符串表达式仅包含非负整数，+， - ，*，/ 四种运算符和空格。整数除法仅保留整数部分。

示例 1:

输入: “3+2*2”
输出: 7

示例 2:

输入: " 3/2 "
输出: 1

示例 3:

输入: " 3+5 / 2 "
输出: 5

说明：

你可以假设所给定的表达式都是有效的。
请不要使用内置的库函数 eval。

题解

没啥复杂的，直接栈模拟，根据优先级先处理乘合除，处理完了再处理加和减。

AC代码

class Solution {
public:
    typedef long long ll;
    vector<ll>str;
    stack<ll>q;
    int calculate(string s) 
    {
        ll base=0;
        for(int i=0;i<s.length();i++)
        {
            if(s[i]==' ')continue;
            if(!(s[i]>='0'&&s[i]<='9'))
            {
                str.push_back(base);
                if(s[i]=='+')
                str.push_back(1e10);
                else if(s[i]=='-')
                str.push_back(2e10);
                else if(s[i]=='*')
                str.push_back(3e10);
                else if(s[i]=='/')
                str.push_back(4e10);
                base=0;
            }
            else
            {
                base*=10;
                base+=(s[i]-'0');              
            }
        }
        str.push_back(base);
        for(int i=0;i<str.size();i++)
        {
            if(q.empty())
            {
                q.push(str[i]);
            }
            else
            {
                if(q.top()==3e10||q.top()==4e10)
                {
                    ll co=q.top();
                    q.pop();
                    ll num=q.top();
                    q.pop();
                    if(co==3e10)
                    {
                        q.push(num*str[i]);
                    }
                    else q.push(num/str[i]);
                }
                else q.push(str[i]);
            }
        }
        str.clear();
        while(!q.empty())
        {
            str.push_back(q.top());
            q.pop();
        }
        for(int i=str.size()-1;i>=0;i--)
        {
            if(q.empty())
            q.push(str[i]);
            else
            {
                if(q.top()==1e10||q.top()==2e10)
                {
                    ll co=q.top();
                    q.pop();
                    ll num=q.top();
                    q.pop();
                    if(co==1e10)
                    {
                        q.push(num+str[i]);
                    }
                    else q.push(num-str[i]);
                }
                else q.push(str[i]);
            }
        }
        return q.top();
    }
};