[HackerRank] Subarray Division

뒤에서 부터 합산하는방식으로 구현하려고했는데, 특정부분이 자꾸 걸려서 앞에서부터 합산하는 방식으로 바꾸었다

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the birthday function below.
def birthday(s, d, m):
    result = 0
    for i in range(0, n-m+1):
        if sum(s[i:i+m]) == d:
            result += 1
    return result
        
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()