49.丑数

49 丑数

1 题目描述

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

示例：

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

2 题目分析

​ 动态规划思想：dp[i]表示第i个丑数，从1开始计数，有dp[1]=1，根据丑数的定义我们知道dp数组中每个元素乘2，乘3，乘5都还是丑数。不妨设置三个指针p2，p3，p5，初始时分别指向位置1，表示当前位置下一个丑数在p2 * 2,p3 * 3,p5 * 5之中，可以想象三个方格，分别存放 * 2，* 3，*5的结果，

​ 由于是三个方格存储的都是全部x2,或者x3,或者x5,因此不可能有重复的出现。当某一个指针的下一个相乘结果加入到dp数组中，则更新它为下一个位置，具体步骤如下：

1. 初始化dp[1] = 1,p2,p3,p5=1
2. 取dp[p2] * 2, dp[p3] * 3, dp[5] * 5的最小值加入到dp数组中，更新被选入的指针++
3. 一直遍历到dp数组大小为n+1时，返回dp[n]

3 代码

public int nthUglyNumber(int n) {
    int[] dp = new int[n + 1];
    int p2 = 1, p3 = 1, p5 = 1;
    dp[1] = 1;

    for (int i = 2; i <= n; i++) {
        int num2 = dp[p2] * 2, num3 = dp[p3] * 3, 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];
}