LeetCode-丑数

每日一题，今天刷到一道感觉比较有意思的题，刚开始按自己的思路解答时发现超时了，看到官方的题解后恍然大悟！！！

题目描述

给你一个整数 n ，请你找出并返回第 n 个 丑数 。

丑数 就是质因子只包含 2、3 和 5 的正整数

示例 1：

输入：n = 10
输出：12
解释：[1, 2, 3, 4, 5, 6, 8, 9, 10, 12] 是由前 10 个丑数组成的序列。

示例 2：

输入：n = 1
输出：1
解释：1 通常被视为丑数。

思路解析

这道题的描述很简单，就是找到第n个丑数，也就是由2，3，5相乘所得的数。本人感觉这道题的难点在于如何快速的求解出结果，寻常方法很容易超时。我最初使用的方法是使用一个set集合来储存已有的丑数，先把1,2,3,4,5放入，如果n的值小于等于5，那就直接返回本身，否则从6开始遍历，遍历的过程中如果该数可以被2，3，5整除且整除后的数在集合中则证明该数也是丑数，当集合大小达到n时，此时就是第n个丑数。这种方式虽然也可以求解，但是时间消耗太大。

题解中给出的时间复杂度较低的方法是使用动态规划，其实我觉得该解法不能完全算是一个动态规划问题，它是将多指针的方法和动态规划相结合，用多个指针实现了动态规划的效果。

那么我们来看解法，可以使用三个指针p2,p3,p5来表示下一个丑数是当前指针指向的丑数乘以对应的质因数。初始时，三个指针的值都是 1。

随后，令 dp[i]=min(dp[p2]×2,dp[p3]×3,dp[p5]×5)，然后再将最小值对应的指针加 1。知道找到第n个数。这实际上就是用三个指针不断地移动来维护dp数组，从而从小到大地找到所有丑数。

也可以理解为三个指针各自指向三个被2，3，5整除且递增的数组，再按大小将三个数组合并。

这种方法的时间复杂度和空间复杂度都是O(n),算是最快的一种方法了。

实现代码

class Solution {
    public int nthUglyNumber(int n) {
        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, 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];
    }
}