use generator to save storage and speed up program
class Solution(object):
def nthSuperUglyNumber(self, n, primes):
"""
:type n: int
:type primes: List[int]
:rtype: int
"""
uglies=[1]
def gen(prime):
for ugly in uglies:
yield ugly*prime
#*unpack the list of generators
merged=heapq.merge(*map(gen,primes))
while len(uglies)<n:
#the first time next() is called, the gen() will be initialized and all elements of primes will run through the generator once and yield one ugly*prime each and pushed to the heap. the other times next() is called, it will return the next smallest element in the heap meanwhile yield one ugly*prime and push onto the heap
ugly=next(merged)
if ugly!=uglies[-1]:
uglies.append(ugly)
return uglies[-1]
313. Super Ugly Number
最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
- 文/潘曉璐 我一進(jìn)店門徐裸，熙熙樓的掌柜王于貴愁眉苦臉地迎上來遣鼓，“玉大人，你說我怎么就攤上這事重贺∑锼睿” “怎么了？”我有些...
- 文/不壞的土叔 我叫張陵气笙，是天一觀的道長次企。 經(jīng)常有香客問我，道長潜圃，這世上最難降的妖魔是什么缸棵？ 我笑而不...
- 正文 為了忘掉前任，我火速辦了婚禮秉犹，結(jié)果婚禮上蛉谜，老公的妹妹穿的比我還像新娘。我一直安慰自己崇堵，他們只是感情好型诚，可當(dāng)我...
- 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著，像睡著了一般沼死。 火紅的嫁衣襯著肌膚如雪轩性。 梳的紋絲不亂的頭發(fā)上，一...
- 文/蒼蘭香墨 我猛地睜開眼，長吁一口氣：“原來是場噩夢啊……” “哼患雇！你這毒婦竟也來了跃脊？” 一聲冷哼從身側(cè)響起，我...
- 序言：老撾萬榮一對情侶失蹤苛吱，失蹤者是張志新（化名）和其女友劉穎酪术，沒想到半個(gè)月后，有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體翠储，經(jīng)...
- 正文 獨(dú)居荒郊野嶺守林人離奇死亡绘雁，尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
- 正文 我和宋清朗相戀三年橡疼，在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片咧七。...
- 正文 年R本政府宣布瘟檩，位于F島的核電站抹缕，受9級特大地震影響，放射性物質(zhì)發(fā)生泄漏墨辛。R本人自食惡果不足惜卓研，卻給世界環(huán)境...
- 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望睹簇。 院中可真熱鬧奏赘，春花似錦、人聲如沸太惠。這莊子的主人今日做“春日...
- 文/蒼蘭香墨 我抬頭看了看天上的太陽凿渊。三九已至梁只，卻和暖如春，著一層夾襖步出監(jiān)牢的瞬間埃脏，已是汗流浹背搪锣。 一陣腳步聲響...
- 正文 我出身青樓，卻偏偏與公主長得像堵幽，于是被迫代替她去往敵國和親旁壮。 傳聞我的和親對象是個(gè)殘疾皇子，可洞房花燭夜當(dāng)晚...
推薦閱讀更多精彩內(nèi)容
- Write a program to find the nthsuper ugly number. Super u...
- Write a program to find the nth super ugly number. Super ...
- Ugly Number II Write a program to find the n-th ugly numb...
