Submission #10044264

---

Source Code Expand

from bisect import *
n, d = map(int, input().split())
x = list(map(int, input().split()))

ans = 0
for i in range(1, n-1):
    left = x[i] - d
    right = x[i] + d
    ans += (i - bisect_left(x, left)) * (bisect_right(x, right) - i - 1)

for i in range(n-2):
    right = x[i] + d
    option = bisect_right(x, right) - i - 1
    ans -= option*(option-1)//2
print(ans)

Submission Info

Judge Result