좌충우돌 IT이야기

(주의) 정답은 올바르게 잘 나오는데 시간초과가 발생합니다. 추후 수정하겠습니다!

문제

N개의 수로 된 수열 A[1], A[2], …, A[N] 이 있다. 이 수열의 i번째 수부터 j번째 수까지의 합 A[i]+A[i+1]+…+A[j-1]+A[j]가 M이 되는 경우의 수를 구하는 프로그램을 작성하시오.

정답

import sys
sys.stdin = open('input.txt', 'rt')
n, m = map(int, input().split())
a = list(map(int, input().split()))
current = 0
cnt = 0
for i in range(len(a)):
    if a[i] == m:
        cnt += 1
    else:
        current = a[i]
        for j in range(i+1, len(a)):
            current += a[j]
            if current == m:
                cnt += 1
                break
            elif current > m:
                break
                
print(cnt)