fork download
copy 
  1. def main():
  2. 	n = int(input())
  3. 	k = int(input())
  4. 	arr = [int(input()) for _ in range(n)]
  5. 	prefix = {}
  6. 	count = 0
  7. 	sum = 0
  8. 	prefix[0] = 1
  9. 	for num in arr:
  10. 		sum+=num
  11. 		if (sum-k) in prefix:
  12. 			count+=prefix[sum-k]
  13.  
  14. 		if sum in prefix:
  15. 			prefix[sum]+=1
  16. 		else:
  17. 			prefix[sum] = 1
  18.  
  19. 	print(count)
  20.  
  21. if __name__=="__main__":
  22. 	main()
Success #stdin #stdout 0.1s 14068KB
comments (?)
stdin
copy 
5
3
1
2
1
1
2
stdout
copy 
3
https://ideone.com/twz7kC
language:
Python 3 (python  3.12)
created:
9 hours 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!