leetcode - 264. Ugly Number II

最新推荐文章于 2024-07-28 09:02:14 发布

Quiteen

最新推荐文章于 2024-07-28 09:02:14 发布

阅读量416

点赞数

分类专栏：算法文章标签： leetcode 动态规划算法

本文链接：https://blog.csdn.net/quiteen/article/details/73064894

版权

算法专栏收录该内容

17 篇文章 0 订阅

订阅专栏

算法系列博客之Dynamic Programming

本篇博客将运用动态规划的思想来解决leetcode上264号问题

问题描述：

Write a program to find the n-th ugly number.
Ugly numbers are positive numbers whose prime factors only include 2, 3, 5. For example, 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 is the sequence of the first 10 ugly numbers.
Note that 1 is typically treated as an ugly number, and n does not exceed 1690.

特意挑出此题发博的原因在于它较之前所写DP问题有明显两处不同
状态转移不依赖于一个给定的特征列表
如果把每个丑数视为一个状态，则它并不是简单的依赖于固定前一项或者前几项

根据基本的合数的质因数分解可以明白这样一个简单的道理
仅需要将前面的丑数简单的乘以2/3/5即可构造出后面的丑数
关键在于丑数的选择以及2/3/5的选择，得出的结果还要做到不重不漏

事实上，考虑它只能通过乘以2/3/5，且每个数最多只能把这三个数各乘一次
问题就可以看作，新的状态依赖于前面的某三个状态，并且这三个状态都一定会顺次后移
不妨用三个数state[3]来记录2/3/5分别对应依赖于哪三个状态
然后用这三个状态构造出三个丑数(即三个新状态)，选择最小的，并将其对应状态顺移

显然5对应状态一定是最靠前的，再前面的状态无需保留
其次还需注意可能两个或者三个状态构造出的三个新状态都相同，这种情况三个状态都应该顺移
至此，可以很清晰的实现代码逻辑结构如下

class Solution(object):
    def nthUglyNumber(self, n):
        if n < 4: return n
        cur = 3
        state = [1, 1, 0]
        numsqueue = [1, 2, 3]
        while cur < n:
            new1 = numsqueue[state[0]] * 2
            new2 = numsqueue[state[1]] * 3
            new3 = numsqueue[0] * 5
            newnum = min(min(new1, new2), new3)
            if newnum == new1:
                state[0] += 1
            if newnum == new2:
                state[1] += 1
            if newnum == new3:
                numsqueue.pop(0)
                state[0] -= 1
                state[1] -= 1
            numsqueue.append(newnum)
            cur += 1
        return numsqueue[-1]