重生之我在代码随想录刷算法第十天 | 150. 逆波兰表达式求值、347.前 K 个高频元素

参考文献链接：代码随想录

本人代码是Java版本的，如有别的版本需要请上代码随想录网站查看。

150. 逆波兰表达式求值

力扣题目链接

解题思路

这道题目也很明显是需要用栈来处理，只要当有运算符的时候，就将栈顶的两个元素拿出来作运算然后再放入栈中。

代码示例

class Solution {
    public int evalRPN(String[] tokens) {
        Stack<Integer> stack = new Stack<>();
        for(int i = 0;i < tokens.length;++i){
            if("+".equals(tokens[i])){
                stack.push(stack.pop() + stack.pop());
            }else if("-".equals(tokens[i])){
                stack.push(-(stack.pop() - stack.pop()));
            }else if("*".equals(tokens[i])){
                stack.push(stack.pop() * stack.pop());
            }else if("/".equals(tokens[i])){
                int first = stack.pop();
                int second = stack.pop();
                stack.push(second / first);
            }else{
                stack.push(Integer.valueOf(tokens[i]));
            }
        }
        return stack.pop();
    }
}

遇到的问题

这道题容易出错的地方就是while循环的条件该不该带等于号，这种情况我一般会拿一组用例在自己脑子里过一遍。

假设{1,2,3}这组用例我们要找的target值是3，第一次循环的左下标是0，右下标是2，找到中间值是2比target值3小，所以我们将左下标调整为中间下标+1即为2，此时我们数组左右下标均为2，但目标值正好是下标2对应的值，所以应有left=right这个循环条件。

347.前 K 个高频元素

力扣题目链接

解题思路

首先看到这道题目要比的是出现次数，那一定能想起来我们以前刷过的哈希表，所以一定先用Map统计次数。然后就是比较次数了，可以用排序算法但是得先把Map转为List很麻烦，所以我们可以用PriorityQueue，也就是堆。

这道题目我们使用小顶堆，把进来的最小元素顶出去，剩下的就是最大的k个。

代码示例

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        int[] res = new int[k];
        Map<Integer,Integer> map = new HashMap<>();
        //先用Map统计次数
        for(int num:nums){
            map.put(num,map.getOrDefault(num,0)+1);
        }
        //PriorityQueue的一个构造方法是传入Comparator，来规定按什么去排序
        PriorityQueue<int[]> queue = new PriorityQueue<>((pair1,pair2)->pair2[1]-pair1[1]);
        //entrySet方法是把map里的东西以一个set集合的形式展现，返回值是Set<Map.Entry<>>
        for(Map.Entry<Integer,Integer> entry:map.entrySet()){
            queue.add(new int[]{entry.getKey(),entry.getValue()});
        }
        for(int i = 0;i<k;i++){
            res[i] = queue.poll()[0];
        }

        return res;
    }
}