牛客刷题---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其他直接入栈即可。

#include <iostream>
#include <stack>
#include <cstdio>
#include <map>
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());
    }
}