結果

問題 No.2260 Adic Sum
ユーザー yabit
提出日時 2023-06-28 12:25:29
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 339 bytes
コンパイル時間 174 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 117,120 KB
最終ジャッジ日時 2024-07-05 05:26:40
合計ジャッジ時間 6,050 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 1 -- * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

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
for i in range(1, ceil(sqrt(M))+1):
    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
print(ans)
0