21/02/27刷题记录Day8

面试题 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;
	}