您的位置:首页 > 编程语言 > Python开发

LintCode python 小白7-超级丑数

2017-07-15 11:09 330 查看
题目:写一个程序来找第 n 个超级丑数。

超级丑数的定义是正整数并且所有的质数因子都在所给定的一个大小为 k 的质数集合内。

比如给你 4 个质数的集合 [2, 7, 13, 19], 那么 [1, 2, 4, 7, 8, 13, 14, 16, 19, 26, 28, 32] 是前 12 个超级丑数。

注意事项

1 永远都是超级丑数不管给的质数集合是什么。

给你的质数集合已经按照升序排列。

0 < k ≤ 100, 0 < n ≤ 10^6, 0 < primes[i] < 1000

样例:给出 n = 6 和质数集合 [2, 7, 13, 19]。第 6 个超级丑数为 13,所以返回 13 作为结果。

题目解析:n代表第n个数字,primes是质因子集合。本题目给定一个质因子集合,找到从1开始升序的第n个超级丑数。其实此题解决可以借鉴丑数||的,只是那道题是已经确定了质因子集合为[2,3,5]。

思路:

1是特殊的超级丑数,即是列表result中的第一个元素。

primes的质数因子集合去乘于列表result,每个质数因子乘于列表中的所有元素,当得到的数大于列表的最后一个元素,则将其放于yy列表中,这样子yy列表中就有len(primes)个,将min(yy)放在result的最后。

循环2,直到result得到第n个数,即result[n-1]。(列表是从0开始)

代码:

class Solution:
# @param {int} n a positive integer
# @param {int[]} primes the given prime list
# @return {int} the nth super ugly number
def nthSuperUglyNumber(self, n, primes):
# Write your code here
#n为第n个数,primes是那个质数集合
result=[1]
if n==1:
return result[0]
while len(result)<n:
yy=[]
for i in range(0,len(primes)):
for j in range(0,len(result)):
if primes[i]*result[j]>result[-1]:
yy.append(primes[i]*result[j])
break
result.append(min(yy))
return result[-1]
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: