【JZ-49】丑数（数学、动态规划）

题目

参考

算法思路

设丑数序列为 $x_1,x_2,x_3,...,x_n$，则下一个丑数 $x_{n+1}$ 为以下三种情况中的最小值：$$x_{n+1}=\{\begin{array}{ll}{x}_a\times 2,& a\in [1,n]\\ x_b\times 3,& b\in [1,n]\\ x_c\times 5,& c\in [1,n]\end{array}$$
即递推公式为 x n + 1 = m i n { x a × 2 , x b × 3 , x c × 5 } x_{n+1}=min\{x_a\times2,x_b\times3,x_c\times5\} xn+1​=min{xa​×2,xb​×3,xc​×5}
其中索引 $a,b,c$ 还需要满足下列条件：

1. $x_a\times 2>x_n\ge x_{a-1}\times 2$，即 $x_a$ 应为首个乘2后大于 $x_n$ 的丑数；
2. $x_b\times 3>x_n\ge x_{b-1}\times 3$，即 $x_b$ 应为首个乘3后大于 $x_n$ 的丑数；
3. $x_c\times 5>x_n\ge x_{c-1}\times 5$，即 $x_c$ 应为首个乘5后大于 $x_n$ 的丑数

具体代码

class Solution {
    public int nthUglyNumber(int n) {
        int a = 0, b = 0, c = 0;
        int[] dp = new int[n];//动态规划列表，dp[i]代表第i+1个丑数
        dp[0] = 1;//第一个丑数为1
        for(int i = 1; i < n; i++){
            int n2 = dp[a] * 2;
            int n3 = dp[b] * 3;
            int n5 = dp[c] * 5;
            dp[i] = Math.min(Math.min(n2, n3), n5);//计算第i+1个丑数
            //更新对应索引
            if(dp[i] == n2)a++;
            if(dp[i] == n3)b++;
            if(dp[i] == n5)c++;
        }
        return dp[n - 1];//返回第n个丑数
    }
}

复杂度分析

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$