結果

問題 No.3456 Common Difference is D
コンテスト
ユーザー LyricalMaestro
提出日時 2026-02-28 20:22:49
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 135 ms / 2,000 ms
コード長 723 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 306 ms
コンパイル使用メモリ 77,888 KB
実行使用メモリ 98,616 KB
最終ジャッジ日時 2026-02-28 20:22:53
合計ジャッジ時間 3,810 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

## https://yukicoder.me/problems/no/3456

from collections import deque

def main():
    N, D = map(int, input().split())
    A = list(map(int, input().split()))

    d_map = {}
    for a in A:
        b = a % D
        if b not in d_map:
            d_map[b] = []
        d_map[b].append((a - b) // D)
    
    answer = 0
    for array in d_map.values():
        dp_map = {}
        for a in array:
            if a not in dp_map:
                dp_map[a] = [0, 0]

            if (a - 1) in dp_map:
                answer += dp_map[a - 1][1]
                dp_map[a][1] += dp_map[a - 1][0]
                
            dp_map[a][0] += 1

    print(answer)





                




if __name__ == "__main__":
    main()
0