牛客刷题---KY102 计算表达式

KY102 计算表达式

知识点:栈
描述:对于一个不存在括号的表达式进行计算
输入描述:存在多组数据,每组数据一行,表达式不存在空格
输出描述:
输出结果
示例1
输入: 6/2+3+3*4
输出:18

本题思路:首先先掌握套路,这种题一定使用的是两个栈(数字栈和符号栈),再加上一个符号级别的map更加方便,这里的符号优先级也可以用其他方法,但map使用更加巧妙。这里使用的是fgets读取,然后赋值给string类型的变量,接着再对这个string类型的字符串遍历,遍历有以下几种情况,1 数字2 符号 3 换行,所以在处理这个字符串之前先加入一个自定义结束符$,直接两种情况即可 1 数字 2 其他,然后再对各种情况详细处理。
注意这个函数的使用:stod()和stoi() 字符串转double int

题目总结:题目不难,这种算是套路性的题目,在刚开始的时候,我是用的int类型的数字栈,测试发现在测试用例中有小数的情况,所以立马使用double类型,最后才通关,在本关和计算器那题不同,就是对于数字之后没有空格,这题直接就是符号或者换行,所以这题还比较好处理一些,先判断是否为数字,再else其他直接入栈即可。

牛客刷题---KY102 计算表达式_第1张图片

#include 
#include 
#include 
#include 
using namespace std;
//题目要求:对于一个不存在括号的表达式进行计算 表达式不存在空格
int main() {
    char buf[1000];
    map<char, int> priority = {
        {'$', 0},
        {'+', 1}, {'-', 1},
        {'*', 2}, {'/', 2}
    };
    while ( fgets(buf, 1000, stdin) != NULL) {
        string expr = buf;
        expr.pop_back();
        expr.push_back('$');
        string num;
        stack<double> numstack;
        stack<char> operstack;

        for (int i = 0; i < expr.size(); i++) {
            if (expr[i] >= '0' && expr[i] <= '9') {
                num.push_back(expr[i]);
            } else {
                numstack.push(stod(num));
                num = "";
//              if(expr[i]=='$'){
//                  if(num!=""){
//                      numstack.push(stoi(num));
//                      num="";
//                  }
//              }
                //6/2+3+3*4=18
                while (!operstack.empty() && priority[operstack.top()] >= priority[expr[i]]) {
                    char oper = operstack.top();
                    operstack.pop();
                    double rhs = numstack.top();
                    numstack.pop();
                    double lhs = numstack.top();
                    numstack.pop();

                    switch (oper) {
                        case '+':
                            numstack.push(lhs + rhs);
                            break;
                        case '-':
                            numstack.push(lhs - rhs);
                            break;
                        case '*':
                            numstack.push(lhs * rhs);
                            break;
                        case '/':
                            numstack.push(lhs / rhs);
                            break;
                    }
                }

                operstack.push(expr[i]);
            }
        }
        printf("%.0lf", numstack.top());
    }
}

你可能感兴趣的:(刷题练习,算法,刷题,牛客,力扣)