fork download
copy 
  1. def main():
  2. 	n = int(input())
  3. 	k = int(input())
  4. 	arr = [int(input()) for _ in range(n)]
  5. 	count = 0
  6. 	mp = {}
  7. 	for i in range(n):
  8. 		comp = k+arr[i]
  9. 		comp2 = arr[i]-k
  10. 		if comp in mp:
  11. 			count+=mp[comp]
  12. 		if k!=0 and comp2 in mp:
  13. 			count+=mp[comp2]
  14. 		if arr[i] in mp:
  15. 			mp[arr[i]]+=1
  16. 		else:
  17. 			mp[arr[i]]=1
  18. 	print(count)
  19. 	return 
  20.  
  21.  
  22. if __name__=="__main__":
  23. 	main()
Success #stdin #stdout 0.07s 14124KB
comments (?)
stdin
copy 
5
2
1
3
5
4
2
stdout
copy 
3
https://ideone.com/DJwZkO
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!