BOJ 2003. 수들의 합2(python)

BOJ 문제 2003. 수들의 합2

문제링크

문제설명

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')

# 투 포인터를 이용해 구현
n, m = map(int, input().split())
lst = list(map(int, input().split()))

# count : 합이 m인 부분 수열의 갯수
count = 0
# interval_sum : 부분 수열의 합
interval_sum = 0
# 부분 수열의 마지막 인덱스
end = 0


# start를 부분 수열의 첫 번째 인덱스로 둔다.
for start in range(n):
    # end를 끝까지 이동시켜가며 만족하는 부분 수열의 합이 있는지 조사
    while interval_sum < m and end < n:
        interval_sum += lst[end]
        end += 1
    # 있다면 초기화 카운트 +1 후 interval_sum 초기화
    if interval_sum == m:
        count+=1
    interval_sum -= lst[start]
print(count)