NYOJ - [第四届河南省程序设计大赛]表达式求值(栈)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/lzyws739307453/article/details/89042924

题目链接:http://nyoj.top/problem/305

• 内存限制:64MB 时间限制:3000ms

题目描述

Dr.Kong设计的机器人卡多掌握了加减法运算以后，最近又学会了一些简单的函数求值，比如，它知道函数min(20,23)的值是20 ，add(10,98) 的值是108等等。经过训练，Dr.Kong设计的机器人卡多甚至会计算一种嵌套的更复杂的表达式。

假设表达式可以简单定义为:
1. 一个正的十进制数 x 是一个表达式。
2. 如果 x 和 y 是 表达式，则 函数min(x,y )也是表达式,其值为x,y 中的最小数。
3. 如果 x 和 y 是 表达式，则 函数max(x,y )也是表达式,其值为x,y 中的最大数。
4．如果 x 和 y 是 表达式，则 函数add(x,y )也是表达式,其值为x,y 之和。

例如，表达式 max(add(1,2),7) 的值为 7。
请你编写程序，对于给定的一组表达式，帮助 Dr.Kong 算出正确答案，以便校对卡多计算的正误。

输入描述

第一行：N 表示要计算的表达式个数(1≤ N ≤ 10)  
接下来有N行，每行是一个字符串，表示待求值的表达式(表达式中不会有多余的空格，每行不超过300个字符，表达式中出现的十进制数都不超过1000)。

输出描述

输出有N行，每一行对应一个表达式的值。

样例输入

3
add(1,2) 
max(1,999) 
add(min(1,1000),add(100,99))  

样例输出

3
999
200

解题思路

用栈做，遇到操作存入操作栈，遇到数字存入数字栈，注意处理数字，遇到')'可以开始运算，将结果再次存入栈中。

#include <bits/stdc++.h>
using namespace std;
int main() {
    int a, b, c, t, l, s, num, wei;
    char str[305];
    scanf("%d", &t);
    while (t--) {
        stack <int> S, A;
        scanf("%s", str);
        l = strlen(str);
        for (int i = 0; i < l; i++) {
            if (str[i] == 'a') {
                S.push(1);
                i += 2; 
            }
            else if (str[i] == 'm') {
                if(str[i + 1] == 'a')
                    S.push(2);
                else S.push(3);
                i += 2; 
            }
            else if (isdigit(str[i])) {
                sscanf(&str[i], "%d%n", &num, &wei);
                A.push(num);
                i += wei - 1;
            }
            else if (str[i] == ')') {
                b = A.top(), A.pop();
                a = A.top(), A.pop();
                c = S.top(), S.pop();
                switch(c) {
                    case 1: s = a + b; break;
                    case 2: s = max(a, b); break;
                    case 3: s = min(a, b); break;
                }
                A.push(s);
            }
        }
        printf("%d\n", A.top());
    }
    return 0;
}