結果
問題 | No.2979 直角三角形の個数 |
ユーザー |
|
提出日時 | 2024-12-03 23:17:13 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 3,380 bytes |
コンパイル時間 | 242 ms |
コンパイル使用メモリ | 82,412 KB |
実行使用メモリ | 141,908 KB |
最終ジャッジ日時 | 2024-12-03 23:17:49 |
合計ジャッジ時間 | 35,842 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 TLE * 1 |
other | AC * 21 WA * 1 TLE * 4 |
ソースコード
from math import isqrt n = int(input()) 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 from itertools import permutations def calc(x,fp): cnt = x l = len(fp) for i in range(l): r = fp[i] if x < r: break cnt -= x//r for i in range(l): r = fp[i] if x < r: break for j in range(i+1,l): s = fp[j] if x < r*s: break cnt += x//(r*s) for i in range(l): r = fp[i] if x < r: break for j in range(i+1,l): s = fp[j] if x < r*s: break for k in range(j+1,l): t = fp[k] if x < r*s*t: break cnt -= x//(r*s*t) for i in range(l): r = fp[i] if x < r: break for j in range(i+1,l): s = fp[j] if x< r*s: break for k in range(j+1,l): t = fp[k] if x < r*s*t: break for g in range(k+1,l): u = fp[g] if x < r*s*t*u: break cnt += x//(r*s*t*u) for i in range(l): r = fp[i] if x < r: break for j in range(i+1,l): s = fp[j] if x < r*s: break for k in range(j+1,l): t = fp[k] if x < r*s*t: break for g in range(k+1,l): u = fp[g] if x < r*s*t*u: break for h in range(g+1,l): v = fp[h] if x < r*s*t*u*v: break cnt -= x//(r*s*t*u*v) return cnt ans = 0 #pの全探索 const = 10**4 for p in range(2,rootn + 1): fp = [] p_ = p while minp[p_] != 0: mp = minp[p_] fp.append(mp) while p_%mp == 0: p_ //= mp # q は奇数 # p < q < 2p # maxの値はm以下 m = n//(2*p) lim = isqrt(m) #ここを境界とします lq = 2 * ((p+1)//2) + 1 rq = min(m+1,2*p) for r in range(lq,rq,2): if m//r <= lim: break if not (0 < r - p < p): continue for i in fp: if r%i == 0: break else: ans += m//r #この倍数まで大丈夫 #この時あきらかに const <= lq olq,orq = lq,rq for res in range(max(m//orq,1), lim+1): lq = m // (res + 1) rq = m // res #加算されるのがlの範囲 lq = max(lq,olq-1) rq = min(rq,orq-1) if lq >= rq: continue # print(res,p,lq,rq) #この内倍数でもなければ2の倍数でもないもの fp_ = [2] + fp while fp_ and fp_[-1] > rq: fp_.pop() assert fp_.count(2) <= 1 cnt = calc(rq,fp_[:]) - calc(lq,fp_[:]) ans += cnt*res print(ans)