給你一個整數(shù) n ，請你找出并返回第 n 個 丑數(shù) 。丑數(shù) 就是只包含質因數(shù)?2、3 和/或?5?的正整數(shù)。示例 1：輸入：n = 10 輸出：12 解釋：[1, 2, 3, 4, 5, 6, 8, 9, 10, 12] 是由前 10 個丑數(shù)組成的序列。 示例 2：輸入：n = 1 輸出：1 解釋：1 通常被視為丑數(shù)。import heapqclass Solution:def nthUglyNumber(self, n: int) -> int:# 1. 起始最先將最小丑數(shù)1放入隊列# 2. 每次從隊列取出最小值x, 然后將x所對應的丑數(shù)2x,3x,5x進入隊列# 3. 對 2 循環(huán)多次，第n次出隊的值就是答案# 為了防止同一丑數(shù)多次進隊，需要使用數(shù)據(jù)結構set來記錄入隊列的丑數(shù)nums = [2,3,5]explored = {1}pq = [1]for i in range(1, n + 1):x = heapq.heappop(pq)if i == n:return xfor num in nums:t = num * xif t not in explored:explored.add(t)heapq.heappush(pq,t)return -1

總結

以上是生活随笔為你收集整理的78. Leetcode 264. 丑数 II (堆-技巧二-多路归并)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。