动态规划：丑数系列

LeetCode263. 丑数

https://leetcode-cn.com/problems/ugly-number/

编写一个程序判断给定的数是否为丑数。

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

输入: 6
输出: true
解释: 6 = 2 × 3

思路

首先判断特殊情况，然后判断num是否能被2或3或5整除，可以整除就用num除以2或3或5，不可以整除就退出循环，最后判断最终结果是否为1，是1为丑数，不是1不为丑数。本题十分简单，仅作为下一题的铺垫

代码

class Solution 
{
public:
    bool isUgly(int num) 
    {
        if (num <= 0)
        {
            return false;
        }

        if (num == 1)
        {
            return true;
        }

        while (true)
        {
            if (num % 2 ==0) num = num / 2;
            else if (num % 3 == 0) num = num / 3;
            else if (num % 5 == 0) num = num / 5;
            else break;
        }

        if (num == 1)
        {
            return true;
        }

        return false;
    }
};

LeetCode264. 丑数II

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

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

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

思路

1.思考状态：找出最优解的性质，明确状态表示什么

dp[i] 表示第i个丑数的值

2.思考状态转移方程：大问题的最有解如何由小问题的最优解得到

首先明确：对于每个丑数来说，它一定是从某个丑数乘以2或3或5得到的。

我们定义三个指针p2、p3、p5，初值都为1，这三个指针分别指向即将被2或3或5乘以的数值
所以：dp[i] = min(dp[p2] * 2, dp[p3] * 3, dp[p5] * 5)
当dp[i] 更新为三者中的最小值后，说明三个指针p2、p3、p5中的某个指针当前工作已完成，于是将其指向下一个数值(即指针所指的数值加1)

3.思考初始状态

1为第一个丑数，所以：dp[1] = 1

4.自底向上计算得到最优解
5.思考是否可以进行空间的优化

代码

class Solution 
{
public:
    //动态规划
    int nthUglyNumber(int n)
    {
        vector<long long> dp(n + 1, 0);
        dp[1] = 1;
        int p2 = 1, p3 = 1, p5 = 1;
        for (int i = 2; i <= n; ++i)
        {
            dp[i] = min(min(dp[p2] * 2, dp[p3] * 3), dp[p5] * 5);

            if (dp[i] == dp[p2] * 2) ++p2;
            if (dp[i] == dp[p3] * 3) ++p3;
            if (dp[i] == dp[p5] * 5) ++p5;
        }

        return dp[n];
    }
};