本题目出自 剑指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, 三个指针 a, b, c 都指向首个丑数。
- 开始循环生成丑数:
a. 计算下一个丑数的候选集 res[a] * 2 , res[b] * 3, res[c] * 5.
b. 选择候选集里面最小的数作为下一个丑数,填入res.
c. 将被选中的丑数对应的指针向后移移动一位. - 最后返回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];
}