洛谷P1981 表达式求值 题解 栈/中缀转后缀

题目链接:https://www.luogu.org/problem/P1981
这道题目就是一道简化的中缀转后缀,因为这里比较简单,只有加号(+)和乘号(*),所以我们只需要开一个存放数值的栈就可以了(如果涉及加减乘除则需要开另一个用于存放符号的栈)。
首先,我们读取一个整数并将其入栈。
然后接下来我们就是一个符号+一个数字这样的读取了。
每次我们读取一个符号和一个数字num:

  • 如果符号是 + ,则判断一下当前栈中有多少元素,如果有超过1个元素,则将栈中所有元素出栈,并将它们的和入栈;
  • 如果符号是 * ,则取出栈顶元素,并且取出来的栈顶元素与num的乘积在入栈。

当输入结束是,栈中的所有元素的乘积就是我们的答案。
实现代码如下:

#include <bits/stdc++.h>
using namespace std;
const int MOD = 10000;
int num;
char c;
stack<int> stk;
int main() {
    scanf("%d", &num);
    num %= MOD;
    stk.push(num);
    while ( (c = getchar()) != EOF ) {
        if (c != '*' && c != '+') break;
        scanf("%d", &num);
        num %= MOD;
        if (c == '+') {
            while (stk.size() > 1) {
                int a = stk.top(); stk.pop();
                int b = stk.top(); stk.pop();
                stk.push( (a + b) % MOD );
            }
            stk.push(num);
        }
        else if (c == '*') {
            int a = stk.top(); stk.pop();
            stk.push(a * num % MOD);
        }
    }
    while (stk.size() > 1) {
        int a = stk.top(); stk.pop();
        int b = stk.top(); stk.pop();
        stk.push( (a + b) % MOD );
    }
    printf("%d\n", stk.top());
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/codedecision/p/11739390.html