【Leetcode】1387. Sort Integers by The Power Value

题目地址：

https://leetcode.com/problems/sort-integers-by-the-power-value/

对于任意数字$x$，若其是偶数则除以$2$，否则乘以$3$再加$1$，经过若干步可以到$1$，所花的步数成为$x$的power。给定一个区间$[lo,hi]$，要求返回这个区间中的power第$k$小的数。关于power的排序，如果两个数power相等，那么谁数值小谁在前。

思路是用堆。先用记忆化搜索计算区间内所有数的power，然后维护一个size为$k$的堆，优先级与上面规定的优先顺序相反，即谁的power大谁优先，若power一样，谁数值大谁优先。遍历数组，如果堆的size不及$k$则直接入堆，否则看一下当前数与堆顶的优先级比较，如果当前数比堆顶的power小，或者power一样但数值小，则将堆顶出堆，将新数放入；否则维持不动。最后返回堆顶即可。代码如下：

import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.PriorityQueue;

public class Solution {
    public int getKth(int lo, int hi, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(1, 0);
    
        for (int i = lo; i <= hi; i++) {
            power(i, map);
        }
        
        // power不同则power大的优先级高，若一样则数值大者优先级高
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>((x, y) -> Objects.equals(map.get(x), map.get(y)) ? -Integer.compare(x, y) : -Integer.compare(map.get(x), map.get(y)));
    
        for (int i = lo; i <= hi; i++) {
            if (maxHeap.size() < k) {
                maxHeap.offer(i);
            } else {
                int top = maxHeap.peek();
                if (map.get(i) < map.get(top) || (map.get(i).equals(map.get(top)) && i < top)) {
                    maxHeap.poll();
                    maxHeap.offer(i);
                }
            }
        }
        
        return maxHeap.peek();
    }
    
    // DFS，并且用map做记忆化
    private int power(int n, Map<Integer, Integer> map) {
        if (map.containsKey(n)) {
            return map.get(n);
        }
        
        if (n % 2 == 0) {
            map.put(n, 1 + power(n / 2, map));
        } else {
            map.put(n, 1 + power(n * 3 + 1, map));
        }
        
        return map.get(n);
    }
}

注解：

算前$k$大第$k$大等问题有一个套路，就是维护size为$k$的堆，但优先级的规定要与题目中排序的优先级相反，题目中要求第$k$小，那么堆就要开最大堆，诸如此类。