剑指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。

方法一（最小堆）

1.解题思路

由于丑数只包含质因子2，3和5，所以我们每次都将前一个丑数的2倍，3倍和5倍加入到最小堆，那么最小堆存储了当前所能拿到的所有丑数，堆顶元素即是下一个丑数。堆里存放的元素会有重复，可以另起一个哈希表去重。
需要注意数值可能超过int范围，所以用long类型存储。

2.代码实现

class Solution {
    public int nthUglyNumber(int n) {
        int[] factors={2,3,5};
        int ugly=0;
        PriorityQueue<Long> queue=new PriorityQueue<>();
        HashSet<Long> set=new HashSet<>();
        queue.offer(1L);
        set.add(1L);
        for(int i=0;i<n;i++){
            long curr=queue.poll();
            ugly=(int)curr;
            for(int fact:factors){
                if(set.add(curr*fact)){
                    queue.offer(curr*fact);
                }
            }
        }
        return ugly;
    }
}

3.复杂度分析

• 时间复杂度：拿到第n个丑数需要进行n次循环，而每次循环最小堆的添加和删除的时间复杂度均为logn，每次总共需要O(logn+log3*n)，所以时间复杂度为O(n*logn)。
• 空间复杂度：需要额外长度为最多为3*n的哈希表和堆 ，所以空间复杂度为O(n)。

方法二（动态规划）

1.解题思路

• 初始化：新建dp数组，用于存放所有的丑数，长度为n，第一个丑数为1
• 推导状态转移方程：当前下标所对应的丑数一定是之前某个丑数的2倍，3倍或5倍，并且是所有可能中最小的那个，现在的问题是怎么记录之前所有可能的最小值，可以使用三个指针，分别指向每一次可能的选择（本次不选，下一次也可能选到，所以能完整列出所有的可能）。所以假设三个指针分别为i2，i3和i5，则状态转移方程为：dp[i]=min(2*dp[i2]，3*dp[i3]，5*dp[i5])。当选中某一个指针对应下标时，指针后移一位。
• 举例说明：(min是每次循环中取的a，b，c三者的最小值)

        1    2    3    4    5    6    8    9    10    12
  
  i     0    1    2    3    4    5    6    7    8     9
  i2    0    1    1    2    3    3    4    4    5     6  
  a          2    4    4    6    8    8    10   10    12
  i3    0    0    1    1    1    2    2    3    3     4
  b          3    3    6    6    6    9    9    12    12
  i5    0    0    0    0    1    1    1    1    2     3
  c          5    5    5    5    10   10   10   10    12
  min   1    2    3    4    5    6    8    9    10    12
  

2.代码实现

class Solution {
    public int nthUglyNumber(int n) {
        int[] dp=new int[n];
        dp[0]=1;
        int i2=0,i3=0,i5=0;
        for(int i=1;i<n;i++){
            int a=2*dp[i2];
            int b=3*dp[i3];
            int c=5*dp[i5];      
            dp[i]=Math.min(Math.min(a,b),c);  
            if(dp[i]==a) i2++;
            if(dp[i]==b) i3++;
            if(dp[i]==c) i5++;
        }
        return dp[n-1];
    }
}

3.复杂度分析

• 时间复杂度：只需进行一次循环，所以时间复杂度为O(n)。
• 空间复杂度：需要额外长度为n的dp数组 ，所以空间复杂度为O(n)。

剑指offer全集入口： 请戳这里