代码随想录算法跟练 | Day11 | 栈与队列 Part02

个人博客主页：http://myblog.nxx.nx.cn
代码GitHub地址：https://github.com/nx-xn2002/Data_Structure.git

Day10

150. 逆波兰表达式求值

题目链接：
https://leetcode.cn/problems/evaluate-reverse-polish-notation/

题目描述：
给你一个字符串数组 tokens ，表示一个根据 逆波兰表示法 表示的算术表达式。

请你计算该表达式。返回一个表示表达式值的整数。

注意：

• 有效的算符为 '+'、'-'、'*' 和 '/' 。
• 每个操作数（运算对象）都可以是一个整数或者另一个表达式。
• 两个整数之间的除法总是 向零截断 。
• 表达式中不含除零运算。
• 输入是一个根据逆波兰表示法表示的算术表达式。
• 答案及所有中间计算结果可以用 32 位 整数表示。

思路：
之前接触过这种题，本身很简单，对于正确的逆波兰表达式，只需要使用一个栈来存储数字，然后当遍历到算符时，将栈顶两个数组取出计算后，将结果压回栈中即可，最后栈中的数就是计算结果。

代码实现：

class Solution {
    public int evalRPN(String[] tokens) {
        Stack<Integer> nums = new Stack<>();
        for (String token : tokens) {
            if (token.equals("+") || token.equals("-") || token.equals("*") || token.equals("/")) {
                int num2 = nums.pop();
                int num1 = nums.pop();
                nums.push(calculate(num1, num2, token));
            } else {
                nums.push(Integer.parseInt(token));
            }
        }
        return nums.pop();
    }

    private static int calculate(int num1, int num2, String operation) {
        if (operation.equals("+")) {
            return num1 + num2;
        } else if (operation.equals("-")) {
            return num1 - num2;
        } else if (operation.equals("*")) {
            return num1 * num2;
        } else if (operation.equals("/")) {
            return num1 / num2;
        }
        return 0;
    }
}

239. 滑动窗口最大值

题目链接：
https://leetcode.cn/problems/sliding-window-maximum/

题目描述：
给你一个整数数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回 滑动窗口中的最大值 。

思路：
因为是求滑动窗口中的最大值，我们只需要维护一个存储了滑动窗口中元素的优先队列即可，然后由题可知，结果数组长度为 n - k + 1，在向队列内添加值的同时，我们需要检查当前的队首元素是否在滑动窗口中，如果当前遍历到了数组中 i 下标的位置，又由滑动窗口长度为 k，我们可以知道，如果元素的下标是在 i 到 i - k 的范围内，它才是有效的，此处，可以使用循环将不符合要求的元素全部出队。

代码实现：

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        int n = nums.length;
        PriorityQueue<int[]> pq = new PriorityQueue<int[]>((a, b) -> {
            return a[0] != b[0] ? b[0] - a[0] : b[1] - a[1];
        });
        for (int i = 0; i < k; ++i) {
            pq.offer(new int[] { nums[i], i });
        }
        int[] ans = new int[n - k + 1];
        ans[0] = pq.peek()[0];
        for (int i = k; i < n; ++i) {
            pq.offer(new int[] { nums[i], i });
            while (pq.peek()[1] <= i - k) {
                pq.poll();
            }
            ans[i - k + 1] = pq.peek()[0];
        }
        return ans;
    }
}

347. 前 K 个高频元素

题目链接：
https://leetcode.cn/problems/top-k-frequent-elements/

题目描述：
给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案

思路：
这道题也是优先队列的应用，相比上一题简单很多，只需要先统计数组中各个元素的出现频率，然后将元素和出现的频率放到优先队列中，按元素出现频率从大到小排序。接下来，将队列前 k 个元素弹出并存储到结果数组里即可。

代码实现：

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        PriorityQueue<int[]> pq = new PriorityQueue<>((item1, item2) -> (item2[1] - item1[1]));
        for (Integer integer : map.keySet()) {
            pq.add(new int[]{integer, map.get(integer)});
        }
        int[] res = new int[k];
        for (int i = 0; i < k; i++) {
            res[i] = pq.poll()[0];
        }
        return res;
    }
}

总结：今天的相对来说也不难，主要是熟悉编程语言里提供的相关栈和队列的api，同时，进一步掌握栈和队列的性质