From f05d888a0b621ca4e99e2b0fb6e23c097006fe41 Mon Sep 17 00:00:00 2001 From: Santo Cariotti Date: Tue, 28 May 2024 10:29:13 +0200 Subject: Init --- progs/dont_care/a242.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 progs/dont_care/a242.py (limited to 'progs/dont_care/a242.py') diff --git a/progs/dont_care/a242.py b/progs/dont_care/a242.py new file mode 100644 index 0000000..dc0ceec --- /dev/null +++ b/progs/dont_care/a242.py @@ -0,0 +1,12 @@ +import heapq +def nth_super_ugly_number(n, primes): + uglies = [1] + def gen(prime): + for ugly in uglies: + yield ugly * prime + merged = heapq.merge(*map(gen, primes)) + while len(uglies) < n: + ugly = next(merged) + if ugly != uglies[-1]: + uglies.append(ugly) + return uglies[-1] \ No newline at end of file -- cgit v1.2.3-18-g5258