본문 바로가기

코딩

[Python] 백준 #2559. 수열

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

 

2559번: 수열

첫째 줄에는 두 개의 정수 N과 K가 한 개의 공백을 사이에 두고 순서대로 주어진다. 첫 번째 정수 N은 온도를 측정한 전체 날짜의 수이다. N은 2 이상 100,000 이하이다. 두 번째 정수 K는 합을 구하기

www.acmicpc.net

 

# 2559 - Python 3

 

import sys

input = sys.stdin.readline

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

arr = list(map(int, input().split()))
pfx = [0]

summ = 0

for i in arr:
    summ += i
    pfx.append(summ)                    # 누적 합 이용

max_temp = pfx[k] - pfx[0]

for i in range(n - k + 1):
    l = i
    r = i + k

    if pfx[r] - pfx[l] > max_temp:
        max_temp = pfx[r] - pfx[l]      

print(max_temp)