[Codetree] 정수 n개의 합 2 Python

https://www.codetree.ai/missions/8/problems/sum-of-n-integers-2/description

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

n, k = map(int, input().split(" "))

num = list(map(int, input().rstrip().split(" ")))
prefix = [0 for _ in range(n)]


# prefix 설정
prefix[0] = num[0]
for i in range(1, n):
    prefix[i] = prefix[i-1] + num[i]

max_val = 0

# 2개 -> 2개 빼기
for i in range(n-k):
    max_val = max(-(prefix[i] - prefix[i+k]), max_val)

print(max_val)