题解 | #牛牛计算器#

牛牛计算器

https://www.nowcoder.com/practice/192ac31d5e054abcaa10b72d9b01cace

该代码使用的编程语言是Java。

该题考察的知识点是利用栈(Stack)数据结构实现后缀表达式的计算。后缀表达式是一种不含括号的数学表达式,例如 "2 3 +" 表示 2 + 3。需要遍历给定的字符串数组 tokens,根据运算符进行相应的计算,并将结果压入栈中,最后返回栈顶元素即可。

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param s string字符串
     * @return int整型
     */
    public int calculate (String s) {
        // write code here
        Stack<Integer> nums = new Stack<>();
        Stack<Character> ops = new Stack<>();
        int num = 0;
        char preOp = '+';
        for (int i = 0; i < s.length(); i++) {
            char c = s.charAt(i);
            if (Character.isDigit(c)) {
                num = num * 10 + c - '0';
            } else if (c == '(') {
                int j = i, cnt = 0;
                for (; i < s.length(); i++) {
                    if (s.charAt(i) == '(') cnt++;
                    if (s.charAt(i) == ')') cnt--;
                    if (cnt == 0) break;
                }
                num = calculate(s.substring(j + 1, i));
            }
            if (!Character.isWhitespace(c) || i == s.length() - 1) {
                if (preOp == '+') {
                    nums.push(num);
                } else if (preOp == '-') {
                    nums.push(-num);
                } else if (preOp == '*') {
                    int temp = nums.pop() * num;
                    nums.push(temp);
                }
                num = 0;
                preOp = c;
            }
        }
        int res = 0;
        while (!nums.isEmpty()) {
            res += nums.pop();
        }
        return res;
    }
}

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务