fork download
copy 
  1. def max_len(arr, k):
  2. 	events = {}
  3.  
  4. 	for x in arr:
  5. 		l = x-k
  6. 		r = x+k
  7.  
  8. 		events[l] = events.get(l, 0)+1
  9. 		events[r+1] = events.get(r+1,0) -1
  10.  
  11. 	current = 0
  12. 	ans = 0
  13.  
  14. 	for x in sorted(events):
  15. 		current += events[x]
  16. 		ans = max(ans, current)
  17. 	return ans
  18.  
  19. t = int(input())
  20.  
  21. for _ in range(t):
  22.     n, k = map(int, input().split())
  23.     a = list(map(int, input().split()))
  24.  
  25.     print(max_len(a, k))
  26.  
Success #stdin #stdout 0.07s 14188KB
comments (?)
stdin
copy 
2
4 0
2 2 5 6
3 1
1 5 6
stdout
copy 
2
2
https://ideone.com/SpSxrV
language:
Python 3 (python  3.12)
created:
2 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!