LeetCode - 264. 丑数 II

题目描述

给你一个整数 n ，请你找出并返回第 n 个 丑数 。
丑数 就是只包含质因数 2、3 和/或 5 的正整数。
1 通常被视为丑数。

题解

一、最小堆
初始时堆为空。首先将最小的丑数 1 加入堆。每次取出堆顶元素 x，则 xx 是堆中最小的丑数，由于 2x, 3x, 5x 也是丑数，因此将 2x, 3x, 5x 加入堆。排除重复元素的情况下，第 n 次从最小堆中取出的元素即为第 n 个丑数。

import heapq
class Solution:
    def nthUglyNumber(self, n: int) -> int:
        heap = []
        heapq.heappush(heap, 1)
        seen = set()
        seen.add(1)
        factors = [2, 3, 5]
        x = 1
        for i in range(n):
            x = heapq.heappop(heap)
            for f in factors:
                y = x * f
                if y not in seen:
                    seen.add(y)
                    heapq.heappush(heap, y)
        return x

二、动态规划
用一个有序数组dp记录前n个丑数。三个指针x2，x3和x5指向dp中的元素，最小的丑数只可能出现在dp[x2]的2倍、dp[x3]的3倍和dp[x5]的5倍三者中间。通过移动三个指针，就能保证生成的丑数是有序的。

 class Solution:
    def nthUglyNumber(self, n: int) -> int:
        dp = [0] * n
        dp[0] = 1
        x2, x3, x5 = 0, 0, 0
        for i in range(1, n):
            dp[i] = min(2 * dp[x2], 3 * dp[x3], 5 * dp[x5])
            if dp[i] == 2 * dp[x2]:
                x2 += 1
            if dp[i] == 3 * dp[x3]:
                x3 += 1
            if dp[i] == 5 * dp[x5]:
                x5+= 1
        return dp[n-1]