Swift.丑数

cubegao 2016-04-01 PM 469℃ 0条
题目描述:把只包含因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含因子7。
习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
import Foundation

class For34Solution {
    func kUglyNum(_ k: Int) -> Int {
        
        if k <= 0 {
            return 0
        }
        
        var uglyNums = [Int]()
        
        uglyNums.append(1)
        var ugly2 = 0
        var ugly3 = 0
        var ugly5 = 0
        var current = 1
        
        
        while current < k {
            
            let minUgly = min(uglyNums[ugly2]*2, uglyNums[ugly3]*3, uglyNums[ugly5]*5)
            uglyNums.append(minUgly)
            
            while uglyNums[ugly2]*2 <= minUgly {
                ugly2 += 1
            }
            while uglyNums[ugly3]*3 <= minUgly {
                ugly3 += 1
            }
            while uglyNums[ugly5]*5 <= minUgly {
                ugly5 += 1
            }
            
            current += 1
        }
        
        return uglyNums[current - 1]
    }
    
    
    func isUgly(_ n: Int) -> Bool {
        if n == 0 {
            return false
        }
        
        var num = n
        
        while num%2 == 0 {
            num /= 2
        }
        
        while num%3 == 0 {
            num /= 3
        }
        
        while num%5 == 0 {
            num /= 5
        }
        
        return num == 1
    }
}

算法思想:最简单的方法就是先通过将一个数不断除以2,3,5来判定该数是不是丑数,而后在从1开始,依次往后判断每个数是不是丑数,并记下丑数的个数,这样当计算的个数为给定值时,便是需要求的第n个丑数,这种方法的时间复杂度为O(k),这里的k为第n个丑数的大小,比如第1500个丑数的大小为859963392,那么就需要判断859963392次,时间效率非常低。
    直观的优化措施就是看能不能将时间复杂度降低到O(n),即只在丑数上花时间,而不在非丑数上浪费时间。剑指offer上给的思路很好,用O(n)的辅助空间来得到O(n)的时间复杂度。其核心思想是:每一个丑数必然是由之前的某个丑数与2,3或5的乘积得到的,这样下一个丑数就用之前的丑数分别乘以2,3,5,找出这三这种最小的并且大于当前最大丑数的值,即为下一个要求的丑数。

github地址:https://github.com/cubegao/LeetCode

标签: 算法

非特殊说明,本博所有文章均为博主原创。

评论啦~