tanmay_newatia

Q5

Oct 29th, 2022
46
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.48 KB | Source Code | 0 0
  1. from collections import defaultdict
  2.  
  3. for _ in range(int(input())):
  4.     n=[int(x) for x in input().split()]
  5.     nl=[int(x) for x in input().split()]
  6.     Sum=n[1]
  7.     prevSum = defaultdict(lambda : 0)
  8.     res = 0
  9.     currsum = 0
  10.     for i in range(0, n[0]):
  11.         currsum += nl[i]
  12.         if currsum == Sum:
  13.             res += 1        
  14.         if (currsum - Sum) in prevSum:
  15.             res += prevSum[currsum - Sum]
  16.         prevSum[currsum] += 1
  17.      
  18.     print(res)
Advertisement
Add Comment
Please, Sign In to add comment