[BOJ] #2559_수열(Python3, 파이썬)

본 게시글은 백준 저지 알고리즘 문제를 '파이썬, Python3' 언어로 풀이한 내용을 주관적으로 정리하였으며, 내용과 관련된 코드리뷰 및 피드백 환영합니다.

티어

Silver III

문제

매일 아침 9시에 학교에서 측정한 온도가 어떤 정수의 수열로 주어졌을 때, 연속적인 며칠 동안의 온도의 합이 가장 큰 값을 알아보고자 
한다.

예를 들어, 아래와 같이 10일 간의 온도가 주어졌을 때, 

3 -2 -4 -9 0 3 7 13 8 -3

모든 연속적인 이틀간의 온도의 합은 아래와 같다.



이때, 온도의 합이 가장 큰 값은 21이다. 

또 다른 예로 위와 같은 온도가 주어졌을 때, 모든 연속적인 5일 간의 온도의 합은 아래와 같으며, 



이때, 온도의 합이 가장 큰 값은 31이다.

매일 측정한 온도가 정수의 수열로 주어졌을 때, 연속적인 며칠 동안의 온도의 합이 가장 큰 값을 계산하는 프로그램을 작성하시오.

입력

첫째 줄에는 두 개의 정수 N과 K가 한 개의 공백을 사이에 두고 순서대로 주어진다. 
첫 번째 정수 N은 온도를 측정한 전체 날짜의 수이다. N은 2 이상 100,000 이하이다. 
두 번째 정수 K는 합을 구하기 위한 연속적인 날짜의 수이다. K는 1과 N 사이의 정수이다. 
둘째 줄에는 매일 측정한 온도를 나타내는 N개의 정수가 빈칸을 사이에 두고 주어진다. 
이 수들은 모두 -100 이상 100 이하이다.

출력

첫째 줄에는 입력되는 온도의 수열에서 연속적인 K일의 온도의 합이 최대가 되는 값을 출력한다.

예제 입력 및 출력 1

10 2
3 -2 -4 -9 0 3 7 13 8 -3

21

예제 입력 및 출력 2

10 5
3 -2 -4 -9 0 3 7 13 8 -3

31

---

풀이

import sys
input = sys.stdin.readline

n, k = map(int, input().split()) # n은 날짜 수 k는 연속적인 날짜 수
tem = list(map(int, input().split())) 
continuous_temp = []
continuous_temp.append(sum(tem[:k]))

for i in range(n - k):
    continuous_temp.append(continuous_temp[i] - tem[i] + tem[i+k])

print(max(continuous_temp))

이 문제의 정답 비율이 낮은 이유는 다들 sum을 여러 번 쓰는 방식을 쓰기 때문인 것 같다.(나도 마찬가지로 처음엔 역시나 시간초과가 떴다...ㅋㅋ) 
이 문제에서 시간 초과가 뜨지 않으려면 sum을 최대한 쓰지 않는 방식으로 풀어야할 것 같아서 위와 같이 풀었다. 
이렇게 하면 범위 안에 있는 값들을 모두 사용하는 것이 아닌 단 세 가지의 값으로 풀 수 있다.