LeetCode 264. 丑数 II

264. 丑数 II

【优先队列】和面试题 17.09. 第 k 个数这个题一样的做法，也是初始化一个只有1的从小到大排列的优先队列（小根堆），每次取出最小值来，然后把最小值跟2，3，5相乘的结果压入最小堆，直到取出过n次数，那么这个数就是答案，需要注意在压入的过程中用一个哈希表来存是否出现过。

class Solution {

    // 优先队列
    // 3:21

    public int nthUglyNumber(int n) {
        int[] arr = new int[] {2, 3, 5};
        PriorityQueue<Long> queue = new PriorityQueue<Long>();
        Set<Long> set = new HashSet();
        queue.offer(1L);
        long ans = 1L;
        for (var i = 0; i < n; i++) {
            ans = queue.poll();
            for (var j = 0; j < 3; j++) {
                long t = arr[j] * ans;
                if (!set.contains(t)) {
                    queue.offer(arr[j] * ans);
                    set.add(t);
                }
            }
        }
        return (int)ans;
    }
}

【归并排序/DP】 

 在刚才那个过程中我们发现如果想得到后面的数，就需要把目前的第i个数出队列，然后分别乘上3,5,7，但是这样得到的数可能是非常往后的数了。那么如果我们想知道第i个数的话，就要知道应该用前面的谁去乘以3,5,7，但是又不能只去记录一个，因为只记录一个的话，这个只乘了3,5,7中的一个就继续了，但是他还没和另外两个乘，所以我们需要三个指针分别记录3,5,7，应该和他们对应的最小的谁去相乘。具体过程为。

初始化第一个元素肯定是1，初始化各个指针都是i3 = i5 = i7 = 1，然后从第二个数开始，就找min（3 * arr[i3], 5 * arr[i5], 7 * arr[i7])，然后看看这个min是由谁乘出来的，就把他的指针向后移动。

class Solution {
    public int getKthMagicNumber(int k) {
        int[] ans = new int[k + 1];
        ans[1] = 1;
        for (int i = 2, i3 = 1, i5 = 1, i7 = 1; i <= k; i++) {
            int a = ans[i3] * 3, b = ans[i5] * 5, c = ans[i7] * 7;
            int min = Math.min(Math.min(a, b), c);
            ans[i] = min;
            if (min == a) i3++;
            if (min == b) i5++;
            if (min == c) i7++;
        }
        return ans[k];
    }
}