def main(k): groups = group(takeLen(k, primes.gen())) trips = (trip for g in groups for trip in combinations(g, 3)) for t in trips: if sequence(*t): rv = int(''.join(map(str, t))) if rv != 148748178147: return rv
def main(lim): for k in count(): for stars in range(1, k): mem = defaultdict(list) for p in takeLen(k, primes.gen()): for key in keys(p, stars): mem[key].append(p) results = [t for t in mem.values() if len(t) >= lim] if results: return min(map(sorted, results))[0]
def ngonalNumsOfLen(n, k): return list(takeLen(k, ngonalNums(n)))