【面试题】丑数（动态规划）

最新推荐文章于 2023-05-15 18:25:16 发布

码农谷阿莫

最新推荐文章于 2023-05-15 18:25:16 发布

阅读量373

点赞数

分类专栏：剑指offer 文章标签： java 动态规划数据结构算法 leetcode

本文链接：https://blog.csdn.net/mr_scx/article/details/107299209

版权

剑指offer 专栏收录该内容

49 篇文章 0 订阅

订阅专栏

题目描述

在这里插入图片描述

解题思路

根据丑数的定义，丑数应该是另⼀个丑数乘以 2、 3或者 5的结果（1除外），满足递推性质。因此我们可以创建⼀个数组，⾥⾯的数字是排好序的丑数，每⼀个丑数都是前⾯的丑数乘以 2、 3或者 5得到的。
在这里插入图片描述

Java

动态规划法

class Solution {
    public int nthUglyNumber(int n) {
        if(n <= 0) return 0;
        int[] dp = new int[n]; // 定义状态表，dp[i]代表第 i + 1 个丑数。
        int a = 0, b = 0, c = 0;
        dp[0] = 1; // 第一个丑数为 1
        for(int i=1; i<dp.length; i++) {
            dp[i] = Math.min(Math.min(dp[a]*2, dp[b]*3), dp[c]*5);
            // 分别判断dp[i]与dp[a]*2，dp[b]*3，dp[c]*5的大小关系，若相等则将对应索引 a,b,c 加1
            // 这里不采用if-else的原因是，存在2*某个丑数 ,3*某个丑数,5*某个丑数相同的情况。当有多个索引满足条件时，均要+1
            if(dp[a]*2 == dp[i]) a++;
            if(dp[b]*3 == dp[i]) b++;
            if(dp[c]*5 == dp[i]) c++;
        }
        return dp[n-1]; // 返回第 n 个丑数
    }
}