Subarray Sums II

def count_subarrays_with_sum(n, x, arr):
    prefix_sum = 0
    prefix_count = {0: 1}  
    count = 0

    for num in arr:
        prefix_sum += num
       
        if prefix_sum - x in prefix_count:
            count += prefix_count[prefix_sum - x]

        if prefix_sum in prefix_count:
            prefix_count[prefix_sum] += 1
            
        else:
            prefix_count[prefix_sum] = 1

    return count

n, x = map(int, input().split())
arr = list(map(int, input().split()))
print(count_subarrays_with_sum(n, x, arr))

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED

Test 22

Verdict: ACCEPTED

Test 23

Verdict: ACCEPTED

Test 24

Verdict: ACCEPTED

Test 25

Verdict: TIME LIMIT EXCEEDED

Test 26

Verdict: ACCEPTED

Test 27

Verdict: ACCEPTED

Test 28

Verdict: ACCEPTED