結果

問題 No.2260 Adic Sum
コンテスト
ユーザー vwxyz
提出日時 2023-04-07 23:00:35
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 759 ms / 2,000 ms
コード長 340 bytes
コンパイル時間 194 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 70,832 KB
最終ジャッジ日時 2024-10-06 07:22:34
合計ジャッジ時間 8,358 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict

N,P=map(int,input().split())
def solve(A):
  if len(A)==1:
    return 0
  dct=defaultdict(list)
  for a in A:
    dct[a%P].append(a)
  ans=0
  for a,A in dct.items():
    le=len(A)
    ans+=solve([(aa-a)//P for aa in A])+le*(le-1)//2
  return ans
A=list(map(int,input().split()))
ans=solve(A)
print(ans)
0