leetcode 264. 丑数 II

发表于 | 分类于

编写一个程序，找出第 n 个丑数。

丑数就是只包含质因数 2, 3, 5 的正整数。

示例:

输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。

说明:

  • 1 是丑数。
  • n 不超过1690。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution(object): # 264. Ugly Number II
	def nthUglyNumber(self, n):
		"""
		:type n: int
		:rtype: int
		"""
		dp = [1 for i in range(n)]
		i2, i3, i5 = 0, 0, 0
		for i in range(1,n):
			n2, n3, n5 = dp[i2]*2, dp[i3]*3, dp[i5]*5
			dp[i] = min(n2, n3, n5)
			if dp[i] == n2: i2 += 1
			if dp[i] == n3: i3 += 1
			if dp[i] == n5: i5 += 1
		# print(dp)
		return dp[-1]