BOJ 2003. 数量の和2(Python)


BOJ問題2003。数の和2


質問リンク
問題の説明
N個の数列A[1],A[2],...,A[N]がある.この数列のiがiからjまでの総和A[i]+A[i+1]+...+A[j-1]+A[j]がMである場合、その数を求めるプログラムを作成します.
I/O例

ソリューション
ダブルポインタで実現しました.始点で端点を移動し、区間を求め、条件を満たすことを確保する
コード#コード#
# 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)