結果
問題 | 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, ceilfrom collections import defaultdictN,P = map(int, input().split())A = list(map(int, input().split()))M = max(A)ans = 0i = 1while pow(P, i) <= M:D = defaultdict(int)p = pow(P, i)for j in range(N):D[A[j] % p] += 1for v in D.values():ans += v*(v-1)//2i += 1print(ans)