結果

問題 No.1514 Squared Matching
コンテスト
ユーザー H3PO4
提出日時 2022-08-11 08:54:41
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 753 bytes
コンパイル時間 181 ms
コンパイル使用メモリ 82,608 KB
実行使用メモリ 818,684 KB
最終ジャッジ日時 2024-09-21 16:59:57
合計ジャッジ時間 5,653 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1 MLE * 1 -- * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

import itertools
from array import array


def isqrt(n):
    x = n
    y = (x + 1) // 2
    while y < x:
        x = y
        y = (x + n // x) // 2
    return x


def primes(n):
    is_prime = array("B", itertools.repeat(1, n + 1))
    is_prime[0] = 0
    is_prime[1] = 0
    for i in range(2, int(n ** 0.5) + 1):
        if not is_prime[i]:
            continue
        for j in range(i * 2, n + 1, i):
            is_prime[j] = 0
    return (i for i in range(n + 1) if is_prime[i])


N = int(input())
table = array("L", range(N + 1))
P = primes(N)
for p in P:
    p2 = p ** 2
    for x in range(p2, N + 1, p2):
        while not table[x] % p2:
            table[x] //= p2

ans = 0
for a in range(1, N + 1):
    ans += isqrt(N // table[a])
print(ans)
0