結果

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

ソースコード

diff #
raw source code 

import sys, math
sys.setrecursionlimit(10**8)
sys.set_int_max_str_digits(0)
INF = 1e18
MOD = 998244353
from bisect import bisect_left, bisect_right
from collections import deque, defaultdict, Counter
from itertools import product, combinations, permutations, groupby, accumulate
from heapq import heapify, heappop, heappush
input = sys.stdin.readline
def I():   return input().rstrip()
def II():  return int(input().rstrip())
def IS():  return input().rstrip().split()
def MII(): return map(int, input().rstrip().split())
def LI():  return list(input().rstrip())
def TII(): return tuple(map(int, input().rstrip().split()))
def LII(): return list(map(int, input().rstrip().split()))
def LSI(): return list(map(str, input().rstrip().split()))
def GMI(): return list(map(lambda x: int(x) - 1, input().rstrip().split()))
def kiriage(a, b): return (a+b-1)//b

N, D = MII()
A = LII()

left, right = defaultdict(int), defaultdict(int)

# 初期状態
left[A[0]] += 1

for i in range(1, N):
    right[A[i]] += 1

ans = 0
for i in range(1, N - 1):
    a = A[i]

    # right から削除
    right[a] -= 1

    # カウント
    ans += left[a - D] * right[a + D]

    # left に追加
    left[a] += 1

print(ans)
0