算法通关村——丑数问题解析

本题目出自 剑指offer49 ，我们把只包含质因子 2、3 和 5 的数称作丑数（Ugly Number），求按从小到大的顺序的第 n 个丑数。

根据丑数的定义，0 和负整数一定不是丑数。

当 n>0 时，若 n 是丑数，则 n 可以写成 n = 2 ^ a * 3 ^ b * 5 ^ c 的形式，其中 a,b,c 都是非负整数。特别地，当 a,b,c 都是 0 时，n=1。

1. 直接计算比较(暴力法)

最容易想到的做法是：为判断 n 是否满足上述形式，可以对 n 反复除以 2,3,5，直到 n 不再包含质因数 2,3,5。若剩下的数等于 1，则说明 n 不包含其他质因数，是丑数；否则，说明n包含其他质因数，不是丑数。

代码实现如下：

    public int getUglyNumber(int index) {
        int num = 0, now = 0;
        while (num < index) {
            now++;
            if (nthUglyNumber1(now)) {
                num++;
            }
        }
        return now;
    }

    public boolean nthUglyNumber1(int index) {
        int[] flag = {2, 3, 5};
        for (int i = 0; i < 3; i++) {
            while (index % flag[i] == 0) {
                index /= flag[i];
            }
        }
        return index == 1;
    }

该方法由于是 暴力遍历，对于较大数据很容易出现超时的问题，所有我们主要来关注一下下面的解法.

2. 动态规划

根据题意，每个丑数都可以由其他较小的丑数通过乘以 222 或 333 或 555 得到。

所以，可以考虑使用一个优先队列保存所有的丑数，每次取出最小的那个，然后乘以 2 , 3 , 5 后放回队列。然而，这样做会出现重复的丑数。例如：

初始化丑数列表 [1]
第一轮： 1 -> 2, 3, 5 ，丑数列表变为 [1, 2, 3, 5]
第二轮： 2 -> 4, 6, 10，丑数列表变为 [1, 2, 3, 4, 6, 10]
第三轮： 3 -> 6, 9, 15 ，出现重复的丑数 6

所以，为了避免重复, 我们可以设置 3 个 指针 a, b, c , 分别表示 下一个丑数是当前指针指向的丑数乘以 2, 3, 5.

利用三个指针生成丑数的算法流程：

1. 初始化丑数列表， 首个丑数为1， 三个指针 a, b, c 都指向首个丑数。
2. 开始循环生成丑数：
   a. 计算下一个丑数的候选集 res[a] * 2 ， res[b] * 3， res[c] * 5.
   b. 选择候选集里面最小的数作为下一个丑数，填入res.
   c. 将被选中的丑数对应的指针向后移移动一位.
3. 最后返回res的最后一个元素即可。
   
   
   
   
   
   （简单用图示演示一下）后面过程省略

代码实现如下：

  public static int getUglyNumber2(int index) {
        int a = 0, b = 0, c = 0;
        int[] res = new int[index];
        res[0] = 1;
        for (int i = 1; i < index; i++) {
            int n1 = res[a] * 2, n2 = res[b] * 3, n3 = res[c] * 5;
            res[i] = Math.min(Math.min(n1, n2), n3);
            if (n1 == res[i])
                a++;
            if (n2 == res[i])
                b++;
            if (n3 == res[i])
                c++;
        }
        return res[index - 1];
    }