結果

問題 No.1058 素敵な数
コンテスト
ユーザー kbys
提出日時 2020-06-02 01:29:30
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 604 bytes
コンパイル時間 169 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 48,952 KB
最終ジャッジ日時 2024-11-22 18:47:31
合計ジャッジ時間 7,337 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other WA * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#1058
import numpy as np
import itertools
def create_prime_list(limit):
    x = limit**0.5
    primes = []
    #print('x={0}'.format(x))
    nums = [x for x in range(2, limit+1)]
    while nums[0]<=x:
        primes.append(nums[0])
        current_prime = nums[0]
        nums = [x for x in nums if x%current_prime != 0]
    primes.extend(nums)
    return primes

N = int(input())

ans = [1]

s = np.array(create_prime_list(10**5+5*10**4))
s = s[s>10**5][:10]
#print(s)

for i,j in itertools.combinations(s, 2):
    ans.append(i*j)
for i in s:
    ans.append(i**2)

ans.sort()
print(ans)

print(ans[N-1])
0