第n小的丑数

每天一道算法题

问题

• 求第 n 小的丑数
• 描述
  • 设计一个算法，找出只含素因子2，3，5 的第 n 小的数。
  • 符合条件的数如：1, 2, 3, 4, 5, 6, 8, 9, 10, 12...
• 挑战
  • 要求时间复杂度为 O(nlogn) 或者 O(n)。

解决

获得一个丑数很容易，因为丑数可以通过 2^x * 3^y * 5^z 获得，同时任意丑数乘以 2、3、5 也是一个丑数。

但是如何获得第 n 小的丑数呢？

当 n 小于等于 6 时，第 n 小的丑数就是 n 本身；但 n 大于 6 时，需要通过前 6 个已知的丑数，从第 7 小的丑数开始计算，一直计算到第 n 小的丑数就能得到结果。

因为任意丑数乘以 2、3、5 也是一个丑数，所以从最小的丑数 1 开始依次乘以 2、3、5 直到恰好大于第 n - 1 的丑数为止*（拿求第 7 小的丑数举例，会得到 [ 8，9，10]）* ，从中取最小值 8 就是第 7 小的丑数；重复上述步骤，一直求到 n 就可以找到第 n 小的丑数。

代码如下

public static int nthUglyNumber(int n) {
     if (n <= 6) {
         return n;
     }
     List<Integer> uglys = new ArrayList<Integer>();
     for (int i = 1; i <= 6; i++) {
         uglys.add(i);
     }
     int p2 = 3, p3 = 2, p5 = 1;

     for (int i = 6; i < n; i++) {
         int lastNumber = uglys.get(i - 1);
         while (uglys.get(p2) * 2 <= lastNumber) p2++;
         while (uglys.get(p3) * 3 <= lastNumber) p3++;
         while (uglys.get(p5) * 5 <= lastNumber) p5++;
 
         uglys.add(Math.min(
             Math.min(uglys.get(p2) * 2, uglys.get(p3) * 3),
             uglys.get(p5) * 5
         ));
      }
 
      return uglys.get(n - 1);
 }

• 为什么代码中 p2、p3、p5 不是从最小的 1 开始(也就是下标0)？
  • 因为第 7 小的丑数是从 [ 2*4，3*3，5*2 ] 中取最小值得来，所以直接将对应的下标3、2、1作为初始值可以减少几次计算。

原文请点击此处