結果
問題 | No.2417 Div Count |
ユーザー |
![]() |
提出日時 | 2023-08-12 14:10:33 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 935 bytes |
コンパイル時間 | 456 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 85,888 KB |
最終ジャッジ日時 | 2024-11-19 17:41:03 |
合計ジャッジ時間 | 3,735 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 7 WA * 34 |
ソースコード
from collections import defaultdict n, k = list(map(int, input().split())) maxA = n-k primeSet = set() primeList = [True] * int(3 + maxA ** 0.5) primeList[1] = False primeList[2] = True primeSet.add(2) for i in range(4, len(primeList), 2): primeList[i] = False for i in range(3, len(primeList)): if primeList[i] == True: for j in range(i*2, len(primeList), i): primeList[j] = False primeSet.add(i) # print(sorted(primeSet)) ans = -1 if k == 0: ans += 1 pFactor = defaultdict(lambda: 0) cnt = 0 while maxA > 1: flag = False # print(pFactor) for p in primeSet: if maxA % p == 0: pFactor[p] += 1 maxA //= p cnt += 1 flag = True continue if cnt == 0: pFactor[maxA] += 1 if not flag: break temp = 1 # print(pFactor) for k in pFactor.keys(): temp *= (1+pFactor[k]) ans += temp print(ans)