結果

問題 No.2260 Adic Sum
コンテスト
ユーザー yabit
提出日時 2023-06-28 12:26:41
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 339 bytes
コンパイル時間 76 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 18,464 KB
最終ジャッジ日時 2024-07-05 05:27:41
合計ジャッジ時間 6,033 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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