面试题 17.09. 第 k 个数
有些数的素因子只有 3,5,7,请设计一个算法找出第 k 个数。注意,不是必须有这些素因子,而是必须不包含其他的素因子。例如,前几个数按顺序应该是 1,3,5,7,9,15,21。
思路
一开始看到题目的时候会以为又是用堆来做的题目。但是仔细一看是丑数相关。和前几天的丑数题目做法一致。
完整代码
public int getKthMagicNumber(int k) {
PriorityQueue<Long> heap = new PriorityQueue<Long>();
HashMap<Long, Integer> hash = new HashMap<Long, Integer>();
heap.add(1L);
hash.put(1L, 1);
int[] factor = new int[] {3,5,7};
long curNum = 1L;
long newNum;
for(int i = 0; i < k; i++) {
curNum = heap.poll();
for (int f : factor) {
newNum = curNum * f;
if(!hash.containsKey(newNum)) {
hash.put(newNum, 1);
heap.add(newNum);
}
}
}
return (int)curNum;
}