結果

問題 No.1058 素敵な数
コンテスト
ユーザー Geckoちゃん
提出日時 2020-05-22 21:57:48
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 31 ms / 2,000 ms
コード長 764 bytes
コンパイル時間 84 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-10-05 17:16:25
合計ジャッジ時間 916 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import sqrt
from collections import Counter

def prime_factorization(n):
    counter = Counter()
    for i in range(2, int(sqrt(n)) + 1):
        while n % i == 0:
            n //= i
            counter[i] += 1

    if n != 1:
        counter[n] += 1

    return list(counter.items())

def prime_factors(n):
    return set(map(lambda x: x[0], prime_factorization(n)))



import itertools

primes = []
cur = 10**5+1
while True:
    now = prime_factors(cur)
    if len(now) == 1:
        primes.append(cur)

    cur += 1
    if len(primes) == 10:
        break
ans = set([1])

#print(primes)
for p1,p2 in itertools.product(primes, repeat=2):
    a = p1*p2
    ans.add(a)
ans = list(ans)
ans.sort() 
#for a in ans:
#    print(a)

print(ans[int(input())-1])
0