看到这题完全没有什么思路啊
学习到一种优秀的题解。
只包含2, 3, 5质因子的数,另一种含义就是,只该数由2,3,5任意的组合乘积可以得到。。。
2,3,5分别用3个索引指示接下来需要乘积的位置。
每次取三个质因子得到的数的最小值(有序的插入到结果集中)
循环判断部分主要是为了排序后,跳过质因子得到相同元素,记住:一定不能if... elif...。每个if都要判断,都有可能进去的。
def GetUglyNumber_Solution(self, index):
# write code here
# 小于0的,直接返回
if index <= 0:
return 0
# 初始化3个索引位置
p2 = p3 = p5 = 0
# 初始化结果集。题目中说,1视为第一个丑数
result = [1]
# 遍历开始 找到除1之外的index-1个丑数
for i in range(1, index):
# 3个中最小的
min1 = min(result[p2]*2, result[p3]*3, result[p5]*5)
# 更新结果集
result.append(min1)
# 更新索引。注意都是if. 3者之间没有什么关联
if result[i] == result[p2]*2:
p2 = p2+1
if result[i] == result[p3]*3:
p3 = p3+1
if result[i] == result[p5]*5:
p5 = p5+1
# 返回第index个丑数
return result[-1]