結果

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

ソースコード

diff #
raw source code 

from collections import deque, defaultdict, Counter
from bisect import bisect_left, bisect_right
from itertools import permutations, combinations, groupby
from heapq import heappop, heappush
import math, sys
input = lambda: sys.stdin.readline().rstrip("\r\n")
def printl(li, sep=" "): print(sep.join(map(str, li)))
def yn(flag): print(Yes if flag else No)
_int = lambda x: int(x)-1
MOD = 998244353 #10**9+7
INF = 1<<60
Yes, No = "Yes", "No"

N, D = map(int, input().split())
A = list(map(int, input().split()))
CNT0 = defaultdict(int)
CNT1 = defaultdict(int)

ans = 0
for i in range(N):
    ans += CNT1[A[i]-D]
    CNT1[A[i]] += CNT0[A[i]-D]
    CNT0[A[i]] += 1
print(ans)
0