LeetCode 面试题 17.09. 第 k 个数

面试题 17.09. 第 k 个数

 

【优先队列】设置一个从小到大的优先队列（小根堆），初始化时将1加入，每次取出最小值，然后把最小值和3，5，7相成的值压入队列，因为可能重复，所以需要用一个哈希表来存储是否出现过，直到取了k次就是最小值，O(klogk)的时间复杂度。

class Solution {

    // 9：30 71
    // 优先队列

    public int getKthMagicNumber(int k) {

        Set<Long> set = new HashSet();
        int[] arr = new int[] {3, 5, 7};
        PriorityQueue<Long> queue = new PriorityQueue();
        long ans = 1L;
        queue.offer(1L);
        for (var i = 0; i < k; i++) {
            ans = queue.poll();
            for (var x: arr) {
                long tmp = ans * x;
                if (!set.contains(tmp)) {
                    set.add(tmp);
                    queue.offer(tmp);
                }
            }
        }
        return (int)ans;
    }
}

【归并排序/DP】我们用三个指针x，y，z来记录当前3，5，7分别应该乘以前面的谁，把三个相乘的结果取最小的min，如果min由某个指针的数和357相乘得到的，就把这个指针++

class Solution {
public:
    int getKthMagicNumber(int k) {
        const int N = k + 1;
        int ans[N];
        ans[1] = 1;
        for (int i = 2, x = 1, y = 1, z = 1; i <= k; i++) {
            int a = 3 * ans[x], b = 5 * ans[y], c = 7 * ans[z];
            int m = min({a, b, c});
            ans[i] = m;
            if (m == a) x++;
            if (m == b) y++;
            if (m == c) z++;
        }
        return ans[k];
    }
};