求第n个丑数

求第n个丑数

题目描述

把只包含质因子2、3和5的数称作丑数（Ugly Number）。例如6、8都是丑数，但14不是，因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数
例如：前20个丑数为：1, 2, 3, 4, 5, 6, 8, 9, 10, 12, 15, 16, 18, 20, 24, 25, 27, 30, 32, 36。

方法一

最容易想到的就是从1开始往上遍历各个整数，然后分别判断其是不是丑数，如果是的话数量加1，不是的话跳过。最后输出第N个丑数就行了。代码如下：

def GetUglyNumber_Solution(index):
    num = 1
    a = 1
    while num != index:
        a += 1
        tmp = a
        while tmp % 2 == 0:
            tmp /= 2
        while tmp % 3 == 0:
            tmp /= 3
        while tmp % 5 == 0:
            tmp /= 5
        if tmp == 1:
            num += 1
    return a

但是这种方法需要对每一个数判断其是否为丑数，很明显效率很低（用这种方法由于运算代价太大，连牛客网上的测试都通过不了）

方法二

由题目的描述可以知道，丑数的一般表达式为 2 i ∗ 3 j ∗ 5 m , ( i , j , m ∈ { 0 , 1 , . . . , N } ) 2^{i}*3^{j}*5^{m},(i,j,m\in\{0,1,...,N\}) 2i∗3j∗5m,(i,j,m∈{0,1,...,N}) 。因此我们可以根据规则生成丑数，然后找到第n个丑数。我们首先可以想到后面的丑数是根据前面的丑数分别乘以2,3,5得到的，但是很显然这样做会有重复的丑数出现。因此这里我会定义三个队列分别存储原丑数乘以2,3,5后的数，然后选取三个队列队头中最小的一个出队列，然后放入丑数列表中，具体例子如下：
t1
丑数列表：1
队列2：2
队列3：3
队列5：5
t2
丑数列表：1,2
队列2：4
队列3：3,6
队列5：5,10
t3
丑数列表：1,2,3
队列2：4,6
队列3：6,9
队列5：5,10,15
t4
丑数列表：1,2,3,4
队列2：6,8
队列3：6,9,12
队列5：5,10,15,20
这里要注意的是当两个队列的队头相同且都为三个队头中的最小值时，两个队列都会过掉这个队头
t5
丑数列表：1,2,3,4,5
队列2：8,10
队列3：9,12,15
队列5：5,10,15,20,25
…
代码如下：

def GetUglyNumber_Solution(self, index):
    # write code here
    if index == 0:
        return 0
    q2 = []
    q3 = []
    q5 = []
    ugly_list = [1]
    while len(ugly_list) < index:
        q2.append(ugly_list[-1]*2)
        q3.append(ugly_list[-1]*3)
        q5.append(ugly_list[-1]*5)
        min_q = min(q2[0],q3[0],q5[0])
        ugly_list.append(min_q)
        if q2[0] == min_q:
            q2 = q2[1:]
        if q3[0] == min_q:
            q3 = q3[1:]
        if q5[0] == min_q:
            q5 = q5[1:]
    return ugly_list[-1]

测试

牛客