题目
有些数的素因子只有 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];
}
}