[BOJ/Python] 백준 2003번 - 수들의 합2

백준 #2003 수들의 합 2

2003번: 수들의 합 2

import sys
input = sys.stdin.readline

n, m = map(int, input().split())   # n: 수열 수, m: 합이 나와야 하는 값
seq = list(map(int, input().split()))

i, j = 0, 0   # 포인터
cnt = 0   # 경우의 수

while i <= j and j <= n:
    now_sum = sum(seq[i:j])

    if now_sum == m:
        cnt += 1
        i += 1
        j += 1
    elif now_sum < m:
        j += 1
    else:
        i += 1

print(cnt)

1. 투포인터를 사용하여 리스트의 구간합을 구한다.
2. 구간합이 m보다 클 경우, 앞에 있는 포인터(i)를 한 칸 뒤로 옮긴다.
3. 구간합이 m보다 작을 경우, 뒤에 있는 포인터(j)를 한 칸 뒤로 옮긴다.

sum을 한 번만 해줬으면 되는데 if문마다 새로 구해서 시간 초과가 계속 났다 😥