【剑指Offer】丑数 解题报告(Python)

标签(空格分隔): 剑指Offer


题目地址:​​https://www.nowcoder.com/ta/coding-interviews​

题目描述:

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

解题方法

第一感觉肯定是把每个数字逐个遍历判断是否是丑数的方式,这样的话效率不高。

比较巧妙的方式是使用空间换时间。我们使用一个数组保存每个丑数,然后生成下一个丑数。

使用了3个指针,分别指向最后一个进行×2,×3,×5操作后会大于当前最大的丑数的位置。那么需要找到下一个丑数的时候,一定会是这三个指针指向的丑数进行对应操作的结果之一。因此,每次都更新三个指针指向下一次操作就会变成最大值的位置。就能一直生成下一丑数。

代码:

# -*- coding:utf-8 -*-
class Solution:
def GetUglyNumber_Solution(self, index):
# write code here
if index < 1:
return 0
res = [1]
t2 = t3 = t5 = 0
nextNum = 1
while nextNum < index:
minNum = min(res[t2] * 2, res[t3] * 3, res[t5] * 5)
res.append(minNum)
if res[t2] * 2 <= minNum:
t2 += 1
if res[t3] * 3 <= minNum:
t3 += 1
if res[t5] * 5 <= minNum:
t5 += 1
nextNum += 1
return res[nextNum - 1]

Date

2018 年 3 月 21 日 – 睡了个懒觉,感觉好幸福。。