题目描述
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
解题思路
(1)最开始考虑的是,将已知的丑数分别乘以2, 3,5,选择出最小的一个加入的ugly队列中。可以使用TreeSet集合,这样既能保证生成的丑数不会重复,而且保证了是从小到大排序的,每次从treeset中选择出第一个元素加入到ugly队列中。这样的方法会耗费很大的时间和空间。
(2)参考牛客网上的高分回答丑数
基本思路是这样的:
丑数一定是由某个较小的丑数乘以2/3/5得到的。因为是按递增排列的,所以选择出最小的一个。
当ugly’中只有数字1时,比较三个数的大小[1 * 2, 1 * 3, 1 * 5],选择2加入ugly中。这时需要比较大小的数字变成了5个【2 ,3,5, 4,6,10】。因为第一次选择的是乘2的结果,所以第二次可定不会再选择1*2作为一个比较对象了。但是注意到一个数p,乘以2/3/5,最小的结果一定是乘以2的,所以只需要拿出2 * 2 = 4来和3和5比较找出最小的值即可。现在的ugly变成了【1,2,3】,下一次需要比较的数字有7个【2 ,3 ,5; 4,6,10; 6,9,15】.
同上一次比较时的想法一样,每一组中选择一个最小的进行比较,即【5,4,6】,选择4.
ugly【1,2,3,4】;【2 ,3 ,5; 4 ,6,10; 6,9,15; 8,12,20】
注意到,【4,6,10】和【8,12,20】虽然是不同数字乘以2/3/5的结果(2和4),但是4是2乘以2得到的,显然这一组数的大小很容易判断的,因此仍然只需要比较三个数【5,6,9】.
观察发现,每次比较的三个数,不一定是最小的三个数,单都是分别属于已知丑数乘以2/3/5的结果。基于此,考虑用三个标记来记录。具体参见代码。
注:丑数的前6个数,分别是【1,2,3,4,5,6】,基于此,才有代码第一个index < 7 的判断。
代码实现
import java.util.ArrayList;
import java.util.TreeSet;
public class Solution {
public int GetUglyNumber_Solution(int index) {
if(index < 7) {
return index;
}
int [] ugly = new int [index];
ugly[0] = 1;
int min = 1;
int _m2 = 0;
int _m3 = 0;
int _m5 = 0;
for(int i = 1; i < index; i++) {
ugly[i] = Math.min(ugly[_m2] * 2,Math.min(ugly[_m3] * 3, ugly[_m5] * 5));
if(ugly[i] == ugly[_m2] * 2)
_m2++;
if(ugly[i] == ugly[_m3] * 3)
_m3++;
if(ugly[i] == ugly[_m5] * 5)
_m5++;
}
return ugly[index-1];
}
}