力扣 面试题 17.09. 第 k 个数

题目

有些数的素因子只有 3，5，7，请设计一个算法找出第 k 个数。注意，不是必须有这些素因子，而是必须不包含其他的素因子。例如，前几个数按顺序应该是 1，3，5，7，9，15，21。

示例

输入: k = 5

输出: 9

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/get-kth-magic-number-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：优先队列（小根堆）

和剑指 Offer 49. 丑数相似

Java实现

class Solution {
    public int getKthMagicNumber(int k) {
        PriorityQueue<Long> pq = new PriorityQueue<>();
        Set<Long> set = new HashSet<>();
        pq.offer(1L);
        set.add(1L);

        int ans = 0;
        for (int i = 0; i < k; i++) {
            long var = pq.poll();
            ans = (int)var;
            if (set.add(var * 3)) pq.offer(var * 3);
            if (set.add(var * 5)) pq.offer(var * 5);
            if (set.add(var * 7)) pq.offer(var * 7);
        }
        return ans;
    }
}

方法2：多路归并

⭐题解：https://leetcode.cn/problems/get-kth-magic-number-lcci/solution/zhua-wa-mou-si-by-muse-77-x2i4/

Java实现

class Solution {
    public int getKthMagicNumber(int k) {
        int idx1 = 0, idx2 = 0, idx3 = 0;
        int[] res = new int[k];
        res[0] = 1;

        for (int i = 1; i < k; i++) {
            int a = res[idx1] * 3;
            int b = res[idx2] * 5;
            int c = res[idx3] * 7;
            int minValue = Math.min(a, Math.min(b, c));
            if (a == minValue) idx1++;
            if (b == minValue) idx2++;
            if (c == minValue) idx3++;
            res[i] = minValue;
        }
        return res[k - 1];
    }
}