题解 | #四则运算#

四则运算

https://www.nowcoder.com/practice/9999764a61484d819056f807d2a91f1e

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        BufferedReader r = new BufferedReader(new InputStreamReader(System.in));
        String a;
        try {
            a = r.readLine();
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        char[] chs = a.toCharArray();
        int i = 0, j, l = chs.length;
        int c1;
        float n1, n2, t = 0f;
        boolean minus;
        while (i < l) {//将所有的中括号、大括号替换为小括号
            if (chs[i] == '[' || chs[i] == '{') chs[i] = '(';
            if (chs[i] == ']' || chs[i] == '}') chs[i] = ')';
            i++;
        }
        Stack<Float> nums = new Stack<>();//初始化栈,存放数字
        Stack<Integer> calcs = new Stack<>();//初始化栈,存放符号
        i = 0;
        while (i < l) {
            if (chs[i] == '(') {//ASCII码40
                calcs.push(40);
                i++;
                continue;
            }
            if (chs[i] == ')') {//ASCII码41
                while ((c1 = calcs.peek()) !=
                        '(') {//遇到反括回,两栈分别弹出运算符和数字进行计算,直到符号栈顶的元素是(;判断栈顶元素,因此用peek。
                    calcs.pop();
                    n2 = nums.pop();//最上面是n2
                    n1 = nums.pop();//其次是n1
                    t = cal(n1, n2, c1);//运算实质是n1?n2
                    nums.push(t);
                }
                calcs.pop();//计算完毕,将小括号弹出
                i++;
                continue;
            }
            if (chs[i] == '+' || chs[i] == '-' || chs[i] == '*' || chs[i] == '/') {
                minus = i > 0 && chs[i - 1] == '(' && chs[i] == '-';
                minus = minus || i == 0 && chs[i] == '-';
                if (minus) nums.push(0f);
                if (calcs.isEmpty()) {//存储括号和运算符的栈是空
                    calcs.push(0 + chs[i]);
                    i++;
                    continue;
                } else {
                    c1 = calcs.peek();
                    if (isHigh(c1,
                               chs[i])) {//栈顶是+-,当前是*/,继续压入栈,栈顶是(,先压入
                        calcs.push(0 + chs[i]);
                    } else {//栈顶和当前都是+-,开始计算;栈顶是*/,先弹出计算
                        c1 = calcs.pop();
                        n2 = nums.pop();
                        n1 = nums.pop();
                        t = cal(n1, n2, c1);
                        nums.push(t);//将计算好的数字存入数字栈
                        i--;//继续判断,符号栈是否是空,直到是空,到if为true的判断,将当前符号压入栈
                    }
                    i++;
                    continue;
                }
            }
            if ((chs[i] - '0' | '9' - chs[i]) >
                    0) {//遇到字符是数字,计算完整数字的大小
                j = i + 1;
                n1 = chs[i] - '0';
                while (j < l) {
                    if ((chs[j] - '0' | '9' - chs[j]) < 0) break;
                    n1 *= 10;
                    n1 += chs[j] - '0';
                    j++;
                }
                nums.push(n1);//将数字压入数字栈
                i = j;
                continue;
            }
            i++;
        }
        while (!calcs.isEmpty()) {//符号栈中还不是空的,逆向计算得到最后结果
            c1 = calcs.pop();
            n2 = nums.pop();
            n1 = nums.pop();
            t = cal(n1, n2, c1);
            nums.push(t);
        }
        t = nums.pop();
        System.out.print(Math.round(t));//结果四舍五入
    }

    //计算方法,43——+,45——-,42——*,47——/,a和b分别为前面的数和后面的数
    private static float cal(float a, float b, int ope) {
        switch (ope) {
            case 43:
                return a + b;
            case 45:
                return a - b;
            case 42:
                return a * b;
            case 47:
                return a / b;
            default:
                return 0;
        }
    }

    //是否高优先级,符号栈顶的元素是+-,当前字符也是+-,可以先计算栈顶的运算符号;
    //符号栈顶的元素是*/,乘除的优先级最高,不管当前运算符是什么,都可以先计算栈顶的运算符号,从数字栈中取值;
    //符号栈顶的元素是+-,当前字符是*/,优先级高于+-,要将当前的运算符压入符号栈;
    //符号栈顶的元素是(,要将当前的运算符压入符号栈
    private static boolean isHigh(int c1, int c2) {
        boolean res = false;
        if (c1 == 43 || c1 == 45) {
            if (c2 == 42 || c2 == 47) res = true;
        } else if (c1 == 40) res = true;
        return res;
    }
}

全部评论

相关推荐

FieldMatching:看成了猪头顾问,不好意思
点赞 评论 收藏
分享
03-18 09:45
莆田学院 golang
牛客749342647号:佬,你这个简历模板是哪个,好好看
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务