題目
把只包含因子2、3和5的數稱作醜數(Ugly Number)。例如6、8都是醜數,但14不是,因爲它包含因子7。 習慣上我們把1當做是第一個醜數。求按從小到大的順序的第N個醜數。
思路
用一個數組記錄當前找到的最小丑數。
代碼
# -*- coding:utf-8 -*-
class Solution:
def GetUglyNumber_Solution(self, index):
# write code here
if index <= 6: return index
pos2 = 0; pos3 = 0; pos5 = 0;
dp = [0 for i in range(index)]
dp[0] = 1
for i in range(1, index):
dp[i] = min(min(dp[pos2] * 2, dp[pos3] * 3), dp[pos5] * 5)
if dp[pos2]*2 == dp[i]: pos2 += 1
if dp[pos3]*3 == dp[i]: pos3 += 1
if dp[pos5]*5 == dp[i]: pos5 += 1
return dp[index - 1]