結果

問題 No.2979 直角三角形の個数
コンテスト
ユーザー lif4635
提出日時 2024-12-03 00:49:47
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 839 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 438 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 148,608 KB
最終ジャッジ日時 2024-12-03 00:50:32
合計ジャッジ時間 44,157 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other AC * 19 TLE * 7
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from functools import cache
from math import isqrt

@cache
def gcd(a,b):
    if b == 0:
        return a
    return gcd(b,a%b)

n = int(input())
#定数倍を考える
# #sqrt(n)以上のとき

rootn = isqrt(n)
minp = [0]*(rootn+1)

for i in range(3,rootn+1,2):
    if minp[i] != 0:
        continue
    for j in range(i,rootn+1,i):
        if minp[j] == 0:
            minp[j] = i

ans = 0
for p in range(2,rootn + 1):
    f = p%2 + 1
    #遇奇の不一致
    fp = []
    p_ = p
    while minp[p_] != 0:
        mp = minp[p_]
        fp.append(mp)
        while p_%mp == 0:
            p_ //= mp
    for q in range(f,p,2):
        r = 2 * p * (p + q)
        if n < r:
            break
        for i in fp:
            if q%i == 0:
                break
        else:
            ans += n//r #この倍数まで大丈夫

print(ans)
0