수들의 합

📚 문제 📚

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

입력 설명

첫째 줄에 N(1 ≤ N ≤ 10,000), M(1 ≤ M ≤ 300,000,000)이 주어진다. 다음 줄에는 A[1], A[2], …, A[N]이 공백으로 분리되어 주어진다. 각각의 A[x]는 30,000을 넘지 않는 자연수이다.

출력 설명

첫째 줄에 경우의 수를 출력한다.

입력 예제

8 3
1 2 1 3 1 1 1 2

출력 예제

5

풀이 📝

N, M = map(int, input().split())
A = list(map(int, input().split()))

lt, rt = 0, 1
tot = A[0]
cnt = 0

while True:
    if tot < M:
        if rt < N:
            tot += A[rt]
            rt += 1
        else:
            break
    elif tot == M:
        cnt += 1
        tot -= A[lt]
        lt += 1
    else:
        tot -= A[lt]
        lt += 1

print(cnt)