HackerRank - Birthday Chocolate

https://www.hackerrank.com/challenges/the-birthday-bar/problem

 

HackerRank - Birthday Chocolate

 

#!/bin/python3
​
import math
import os
import random
import re
import sys
​
​
def birthday(s, d, m):
    count = 0
    for i in range(0, len(s)-m+1):
        if(int(sum(s[i:i+m])) == d):
            count += 1
    return count
​
​
if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')
​
    n = int(input().strip())
​
    s = list(map(int, input().rstrip().split()))
​
    dm = input().rstrip().split()
​
    d = int(dm[0])
​
    m = int(dm[1])
​
    result = birthday(s, d, m)
​
    fptr.write(str(result) + '\n')
​
    fptr.close()
​