SubarrayDivision.py

#!/bin/python3

import math
import os
import random
import re
import sys

#
# Complete the 'birthday' function below.
#
# The function is expected to return an INTEGER.
# The function accepts following parameters:
#  1. INTEGER_ARRAY s
#  2. INTEGER d
#  3. INTEGER m
#

def birthday(s, d, m):
    # Write your code here
    start = 0
    end = start + m - 1
    
    i = start
    sum = 0
    count = 0
    while i <= end:
        sum = sum + s[i]
        i = i + 1
    
    if sum == d:
        count = count + 1
    
    start = start + 1
    end = end + 1
    
    n = len(s)
    while end < n:
        sum = sum - s[start - 1] + s[end]
        if sum == d:
            count = count + 1
        start = start + 1
        end = end + 1
    
    return count
          

if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    n = int(input().strip())

    s = list(map(int, input().rstrip().split()))

    first_multiple_input = input().rstrip().split()

    d = int(first_multiple_input[0])

    m = int(first_multiple_input[1])

    result = birthday(s, d, m)

    fptr.write(str(result) + '\n')

    fptr.close()