剑指 Offer 49题：丑数

1.题目描述

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

示例1：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/chou-shu-lcof/

2.思路分析

丑数的递推性质： 丑数只包含因子 2, 3, 5 ，因此有 “丑数 = 某较小丑数 × 某因子” （例如：10 = 5 × 2）。

定义dp数组： 设动态规划列表 dp ，dp[i]表示第 i 个丑数

初始值：dp[1] = 1

状态转移方程：

3.解答 

class Solution {
    //DP

    public int nthUglyNumber(int n) {
        //定义dp数组：dp[i]表示第i个丑数
        int[] dp = new int[n + 1];
        //初始化
        dp[1] = 1;

        //定义三个指针，表示丑数
        int p2 = 1, p3 = 1, p5 = 1;

        //状态转移
        for(int i = 2; i <= n; i++){
            int num2 = dp[p2] * 2;
            int num3 = dp[p3] * 3;
            int num5 = dp[p5] * 5;
            
            //按大小顺序添加丑数
            dp[i] = Math.min(Math.min(num2, num3), num5);

            //更新指针
            if(dp[i] == num2){
                p2++;
            }
            if(dp[i] == num3){
                p3++;
            }
            if(dp[i] == num5){
                p5++;
            }
        }

        return dp[n];
    }
}

时间复杂度 O(N) ： 其中 N = n ，动态规划需遍历计算 dp 列表。
空间复杂度 O(N)： 长度为 N 的 dp 列表使用 O(N) 的额外空间。