結果
問題 | No.2260 Adic Sum |
ユーザー |
|
提出日時 | 2023-06-28 12:27:54 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 436 ms / 2,000 ms |
コード長 | 342 bytes |
コンパイル時間 | 162 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 178,972 KB |
最終ジャッジ日時 | 2024-07-05 05:29:00 |
合計ジャッジ時間 | 5,838 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
from math import sqrt, ceil from collections import defaultdict N,P = map(int, input().split()) A = list(map(int, input().split())) M = max(A) ans = 0 i = 1 while pow(P, i) <= M: D = defaultdict(int) p = pow(P, i) for j in range(N): D[A[j] % p] += 1 for v in D.values(): ans += v*(v-1)//2 i += 1 print(ans)