【LeetCode每日一题】2022-09-28 面试题 17.09. 第 k 个数 Java实现

要得到从小到大的第 k 个数，可以使用最小堆实现。

初始时堆为空。首先将最小的数 1 加入堆。

每次取出堆顶元素 x，则 x 是堆中最小的数，由于 3x, 5x, 7x 也是符合要求的数，因此将 3x, 5x, 7x加入堆。

但是，上述做法会导致堆中出现重复元素的情况。为了避免重复元素，可以使用哈希集合去重，避免相同元素多次加入堆。

在排除重复元素的情况下，第 k 次从最小堆中取出的元素即为第 k 个数

代码

class Solution {
    public int getKthMagicNumber(int k) {
        int[] factors = {3, 5, 7};
        Set<Long> seen = new HashSet<Long>();
        PriorityQueue<Long> heap = new PriorityQueue<Long>();
        seen.add(1L);
        heap.offer(1L);
        int magic = 0;
        for (int i = 0; i < k; i++) {
            long curr = heap.poll();
            magic = (int) curr;
            for (int factor : factors) {
                long next = curr * factor;
                if (seen.add(next)) {
                    heap.offer(next);
                }
            }
        }
        return magic;
    }
}

但是，当弹出第k个数的时候，优先队列里还是有数字的，然而，这些数字并没有被利用到，所以，还有改进的空间。