剑指offer系列之三十二：寻找丑数

题目描述

把只包含因子2、3和5的数称作丑数（Ugly Number）。例如6、8都是丑数，但14不是，因为它包含因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。

因为丑数只有2、3和5这三个因子，那么如果一个是丑数的话，一定是可以被这个三个因子整除的，所以我们可以通过把一个数一直被三个因子除，这样最后如果该数变成1的话（因为第一个丑数是1），那么就验证了该数就是丑数。那么如果想找出第k个丑数的话，就可以从第一个数开始循环，对每一个数进行丑数的判断，从而找到符合要求的第k个丑数。下面是这种思路的代码实现：

public int GetUglyNumber_Solution2(int index) {
        if (index <= 0)
            return 0;
        int number = 0;
        int uglyNumCount = 0;
        while (uglyNumCount < index) {
            number++;
            if (isUglyNumber(number)) {
                uglyNumCount++;
            }
        }
        return number;
    }

    private boolean isUglyNumber(int number) {
        while (number % 2 == 0)
            number /= 2;
        while (number % 3 == 0)
            number /= 3;
        while (number % 5 == 0)
            number /= 5;
        return (number == 1) ? true : false;
    }

这种算法在我提交的时候出现了运行超时的情况，仔细分析发现，我们队每一个数都做了丑数的判断，而且判断一个丑数需要经历三个循环，这是导致性能瓶颈的关键，那么有没有可能使用其他的方法判断呢？我们可以维护一个从小到大的丑数列表，每次得到一个新的丑数之后，就更新这个列表，这个列表的更新思路是：因为丑数只有以上三个因子，所以在排序的丑数列表中，后面的某个丑数一定是前面某个丑数乘以2或3或5的结果，所以我们可以以这三个因子得到不同的丑数，然后把乘以2或3或5的得到的丑数进行排序，把最小的丑数更新到丑数列表中；如果列表的某个丑数乘以2或3或5的结果刚好是丑数中的最小值，那么就继续把该数对应的下标增加1，表示继续寻找下一个新的丑数。

下面是基于这种思路的实现代码（已被牛客AC）：

public int GetUglyNumber_Solution(int index) {
        if (index <= 0)
            return 0;
        int[] uglyNumbers = new int[index];
        uglyNumbers[0] = 1;
        int nextUglyNumberIndex = 1;
        int multiplyNumber2 = 0;
        int multiplyNumber3 = 0;
        int multiplyNumber5 = 0;
        while (nextUglyNumberIndex < index) {
            int min = min(uglyNumbers[multiplyNumber2] * 2,
                    uglyNumbers[multiplyNumber3] * 3,
                    uglyNumbers[multiplyNumber5] * 5);
            uglyNumbers[nextUglyNumberIndex] = min;
            nextUglyNumberIndex++;
            if (uglyNumbers[multiplyNumber2] * 2 == min)
                ++multiplyNumber2;
            if (uglyNumbers[multiplyNumber3] * 3 == min)
                ++multiplyNumber3;
            if (uglyNumbers[multiplyNumber5] * 5 == min)
                ++multiplyNumber5;

        }
        return uglyNumbers[nextUglyNumberIndex - 1];
    }

    private int min(int multiplyNumber2, int multiplyNumber3,
            int multiplyNumber5) {
        int min = (multiplyNumber2 < multiplyNumber3) ? multiplyNumber2
                : multiplyNumber3;
        return (min < multiplyNumber5) ? min : multiplyNumber5;
    }