leetcode:剑指 Offer 49. 丑数，动态规划

题目描述

我们把只包含质因子 2、3 和 5 的数称作丑数（Ugly Number）。求按从小到大的顺序的第 n 个丑数

示例：

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

解题思路：
每个丑数都是由2，3，5相乘得到的，如果已经得到$n-1$个数，则第$n$个数可以由前面的数得到，并且是有一定规律的，本题属于动态规划。
我们即第n个数为$L_n$，则有
$L_n=min(L_a\ast 2,L_b\ast 3,L_c\ast 5),a,b,c\le n-1$
且$a,b,c$满足
$$f(x)=\{\begin{array}{r}{L}_a\times 2>x_n\ge L_{a-1}\times 2\\ L_b\times 3>x_n\ge L_{b-1}\times 3\\ L_c\times 5>x_n\ge L_{c-1}\times 5\end{array}$$
也就是$L_a\times 2$是第一个乘$2$之后大于$L_n$的丑数，所以，$a$可以按递增的顺序，同理$b,c$也可按递增的顺序得到。

假设：
状态定义，设动态规划列表为 $L_n$，$L_i$表示第$i+1$个丑数。索引$a,b,c$指向$L_n$第1个数，$L_0=1$
转移方程： 每次得到$L_i$之后，索引$a,b,c$都需要更新。$L_i$由$L_a$，$L_b$，$L_c$哪个得到，则相应的下标$+1$，指向下一个数。
初始状态： $L_0=1$
返回值： $L_{n-1}$

代码:

class Solution(object):
    def nthUglyNumber(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n == 1: return 1
        listn, a, b, c = [1] * n, 0, 0, 0
        for i in range(1, n):
            n2, n3, n5 = listn[a] * 2, listn[b] * 3, listn[c] * 5
            listn[i] =  min(n2, n3, n5)
            if listn[i] == n2: a += 1
            if listn[i] == n3: b += 1
            if listn[i] == n5: c += 1
        # print(listn)
        return listn[i]