剑指offer 丑数

题目

把只包含因子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]