LeetCode 264 && 剑指Offer 49. Ugly Numbers II 丑数 II

题目描述

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

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

示例:

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

1 是丑数。
n 不超过1690。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/ugly-number-ii

思路

• 采用动态规划思想，假设要找下标i对应的丑数dp[i]，可以用i之前的所有丑数乘若干个2直到大于上一个丑数dp[i-1]，记此数为num1；同理用i之前的所有丑数乘若干个3直到大于上一个丑数dp[i-1]，记此数为num2；用i之前的所有丑数乘若干个5直到大于上一个丑数dp[i-1]，记此数为num3。这三个数中的最小数字就是第i个丑数dp[i]。
• 但是呢，其实没必要把i之前的所有丑数乘2或者乘3或者乘5。在i之前的丑数中，肯定存在一个丑数（下标记为index2），乘2以后正好大于i的上一个丑数dp[i-1],index2之前的丑数乘2都小于等于dp[i-1]；我们只需要记录index2，每次直接用这个下标对应的数乘2就行，并且在下标不满足时更新下标。同理我们也要记录乘3和乘5对应的下标。

c++

1->2->3->4->5->6->8->9->10···

class Solution {
public:
    int nthUglyNumber(int n) {
        vector<int>dp(n+1);
        dp[1]=1;
        int index2=1; //此下标对应的数乘2刚好大于上一个丑数，此下标之前的数乘2都小于等于上一个丑数
        int index3=1; //类似上方，乘2变成乘3
        int index5=1; //类似上方，乘3变成乘5
        for(int i=2;i<=n;++i)
        {
            dp[i]=min(min(dp[index2]*2,dp[index3]*3),dp[index5]*5);
            while(dp[index2]*2<=dp[i]) ++index2;
            while(dp[index3]*3<=dp[i]) ++index3;
            while(dp[index5]*5<=dp[i]) ++index5;
        }
        return dp[n];
    }
};