LeetCode刷题051 丑数

最新推荐文章于 2024-06-12 10:00:23 发布

ZHOUYC1024

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量35

点赞数

分类专栏： LeetCode刷题文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/ZHOUYC1024/article/details/129333414

版权

LeetCode刷题专栏收录该内容

60 篇文章 0 订阅

订阅专栏

剑指 Offer 49. 丑数

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

示例:

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

说明:

1 是丑数。
n 不超过1690。

题解

Code

如果一个数是丑数，则其乘2，乘3，乘5得到的数也是丑数。
主要是如何判断重复数字。
堆与集合的解法：

class Solution {

    PriorityQueue<Long> heap = new PriorityQueue<Long>();
    HashSet<Long> set = new HashSet<Long>();

    public int nthUglyNumber(int n) {
        if(n <= 0 ){
            return -1;
        }
        heap.add(1l);
        set.add(1l);
        int result = 0;
        int[] arr = new int[] {2, 3, 5};
        while(n-- > 0){
            long x = heap.poll();
            result = (int) x;
            for(int i : arr){
                long t = i * x;
                if(!set.contains(t)){
                    heap.add(t);
                    set.add(t);
                }
            }
        }
        return result;
    }
}

动态规划解法：
dp[i] 表示第i个丑数
dp[px] * x 表示从dp[px]乘x 得到的大于dp[i - 1]的最小数，则第i个丑数为他们的最小值。取得最小值的px的指针应当自增。
如果dp[px]为丑数，则dp[px] * x 也是丑数

class Solution {
    public int nthUglyNumber(int n) {
        if(n <= 0 ){
            return -1;
        }
        int[] dp = new int[n + 1];
        dp[1] = 1;
        int p2 = 1, p3 = 1, p5 = 1;
        for(int i = 2; i <= n; i++){
            dp[i] = Math.min(dp[p2] * 2, Math.min(dp[p3] * 3, dp[p5] * 5));
            if(dp[i] == dp[p2] * 2)
                p2++;
            if(dp[i] == dp[p3] * 3)
                p3++;
            if(dp[i] == dp[p5] * 5)
                p5++;
        }
        return dp[n];
    }
}