[백준][Python] 2559 수열

정답이 음수값인 케이스를 고려못해서 한참 고생함.. max값 초기화는 늘 음의 무한대로 할것.. 0으로 햇다가 낭패봄

 

import sys
input = sys.stdin.readline


N, K = map(int, input().split())
arr = list(map(int, input().split()))

left = 0
sum_arr = 0
max_sum = -float('inf')
for right in range(N):
    sum_arr += arr[right]
    if right-left+1 == K:
        max_sum = max(max_sum,sum_arr)
        sum_arr -= arr[left]
        left += 1
print(max_sum)