結果
問題 | No.2260 Adic Sum |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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)