fork download
copy 
  1. # your code goes here
  2.  
  3. def pfx(arr, n):
  4. 	pfx_dict = {}
  5. 	pfx_sum = [0]*(n)
  6. 	pfx_sum[0] = arr[0]
  7. 	for i in range(1, n):
  8. 		pfx_sum[i] = pfx_sum[i-1]+arr[i]
  9.  
  10. 	for n in pfx_sum:
  11. 		pfx_dict[n] = pfx_dict.get(n, 0)+1
  12.  
  13. 	return pfx_sum, pfx_dict
  14.  
  15.  
  16. def func(arr, n, k):
  17. 	pfx_sum, pfx_dict = pfx(arr, n)
  18.  
  19. 	count = 0
  20.  
  21. 	for i in range(n):
  22. 		j = pfx_sum[i]-k
  23. 		count += pfx_dict.get(j, 0)
  24.  
  25. 	return count
Success #stdin #stdout 0.08s 14204KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
Standard output is empty
https://ideone.com/g1UErg
language:
Python 3 (python  3.12)
created:
10 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!