[백준] 2003번(수들의 합 2) with Python

문제

www.acmicpc.net/problem/2003

2003번: 수들의 합 2

풀이

시간초과가 나서 투포인터를 공부해 문제를 풀어보았다.

합이 M보다 작은 경우 end 1 증가 시키고, M과 같으면 카운트, M보다 커지면 start 값을 1 증가시킨다. (반복) 

답

N, M = map(int, input().split(' '))
A = list(map(int, input().split(' ')))
cnt = 0
x = 0
end = 0

for start in range(N) :
    while x < M and end < N :
        x += A[end]
        end += 1
    if x == M :
        cnt += 1
    x -= A[start]

print(cnt)

 

N, M = map(int, input().split(' '))
A = list(map(int, input().split(' ')))
cnt = 0
j = 1

for i in range(N) :
  if A[i] == M :
    cnt += 1
  elif A[i] < M and j > i :
    x = A[i]
    for j in range(1,N) :
      x += A[j]
      if x == M :
        cnt += 1
  else :
    pass

print(cnt)

시간초과가 나왔던 코드