丑数2

264. 丑数 II

难度中等504

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

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

示例:

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

说明:

1. 1 是丑数。
2. n 不超过1690。

通过次数48,845

提交次数87,743

动态规划

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];
    }
}

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/ugly-number-ii/solution/chou-shu-ii-by-leetcode-solution-uoqd/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

堆

class Solution {
    public int nthUglyNumber(int n) {
        int primes[] =new int[]{2,3,5};
        int[] ugly = new int[n];
        ugly[0] = 1;
        PriorityQueue<int[]> pq = new PriorityQueue<>((o1, o2) -> o1[0]-o2[0]);//lambda表达式。以数组的第一个元素升序排序。
        // 把原来的加进去
        for(int i=0;i<primes.length;i++) pq.add(new int[]{primes[i],0,i});
        
        for (int i = 1; i < n; i++) {
            int[] tmp = pq.poll();
            // tmp[0]存储的是当前的丑数
            ugly[i] = tmp[0];
            // tmp[1]存储的是从哪个丑数计算而来,tmp[2]存储的是从那个质数因子计算而来
            // 把原先的丑数后移一个
            tmp[0] = ugly[++tmp[1]]*primes[tmp[2]];//tmp[1]和tmp[2]分别为丑数和质因子的索引。
            pq.add(tmp);
            if(ugly[i]==ugly[i-1]) i--;//如果ugly[i]和ugly[i-1]重复了，那就再重新计算一次ugly[i]。
        }
        return ugly[n - 1];
    }
}

赏

谢谢你请我吃糖果

微信