[백준] 2003. 수들의 합 2 / 파이썬(python)

문제

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을 넘지 않는 자연수이다.

출력

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

https://www.acmicpc.net/problem/2003

2003번: 수들의 합 2

---

문제 풀이

 

시간 초과된 풀이)

출력값은 정상적이나 시간 초과.

n, m = map(int, input().split())

seq = list(map(int, input().split()))
cnt = 0

for i in range(n):
    for j in range(i+1, n+1):
        if sum(seq[i:j]) > m:
            continue
        if sum(seq[i:j]) == m:
            cnt += 1
        
print(cnt)

 

정답 풀이)

 

Two Pointer 기법을 활용한 풀이이다. 

시작점(start)과 끝점(end)를 지정하여 모든 경우의 수에 대해 탐색한다.

n, m = map(int, input().split())
seq = list(map(int, input().split()))

start, end = 0, 1
cnt = 0

while end <= n and start <= end:
    sum_seq = sum(seq[start : end])

    if sum_seq < m: 
        end += 1
    elif sum_seq == m:
        cnt += 1
        end += 1
    else:  ## sum_seq > m 인 경우
        start += 1 

print(cnt)