【剑指offer】丑数 -- Python 实现

【题目描述】
把只包含质因子2、3和5的数称作丑数（Ugly Number）。例如6、8都是丑数，但14不是，因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
【解题思路】
按照丑数的大小一个个的插入丑数，这样可以以O(n)的时间复杂度实现。设置3个指针pos_2, pos_3, pos_5，分别指向要乘以2、3、5的位置。分别让这三个指针位置的数字乘以2、3、5，将最小的结果加入到丑数序列中，如果是pos_2位置的数乘以2最小，证明插入的是pos_2位置的数乘以2，则pos_2往前移一位；如果是pos_3位置的数乘以3最小，证明插入的是pos_3位置的数乘以3，则pos_3往前移一位；如果是pos_5位置的数乘以5最小，证明插入的是pos_5位置的数乘以5，则pos_5往前移一位。用Python实现的代码如下：

# -*- coding:utf-8 -*-
class Solution:
    def GetUglyNumber_Solution(self, index):
        # write code here
        if index == 0:
            return 0
        res = [1]
        pos_2, pos_3, pos_5 = 0, 0, 0
        for i in range(index):
            insert_item = min(res[pos_2]*2, res[pos_3]*3, res[pos_5]*5)
            res.append(insert_item)
            if insert_item == res[pos_2] * 2:
                pos_2 += 1
            if insert_item == res[pos_3] * 3:
                pos_3 += 1
            if insert_item == res[pos_5] * 5:
                pos_5 += 1
        return res[index - 1]