백준 2559 : 수열 (Python)

문제

2559번: 수열 (acmicpc.net)

2559번: 수열

 

---

문제

매일 아침 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

예제 출력 1 복사

21

예제 입력 2 복사

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

예제 출력 2 복사

31

출처

Olympiad > 한국정보올림피아드 > KOI 2011 > 초등부 1번

• 잘못된 데이터를 찾은 사람: tncks0121

알고리즘 분류

• 누적 합
• 두 포인터

 

---

import sys
input = sys.stdin.readline

result =[]

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

for i in range(n - k + 1):
    result.append(sum(a[i:i+k]))
        
print(max(result))

그냥 input()을 통해 값을 받는 것보다 sys.stdin.readline()이 더 빠르단 사실을 안 이후로는 대부분 저렇게 형식을 바꾸어 사용하고 있습니다.

입력 받을 개수 n과 받은 값들을 k개 만큼 연속적으로 합하여 가장 큰 값을 찾아냅니다.

 

for 문에서 n개를 k개로 연속적인 조합으로 만들 수 있는 수는 n - k +1 회입니다 따라서 해당 회차만큼 result 리스트를 만들어 범위에 대한 합값을 저장 후 최대 값을 출력하였지만 범위 안의 값을 모두 더하는데 있어서 시간이 오래 걸려 시간 초과가 출력 되었습니다.

 

import sys
input = sys.stdin.readline

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

result = []
result.append(sum(a[:k]))

for i in range(n - k):
    result.append(result[i] - a[i] + a[k+i])
        
print(max(result))

지나치게 많은 sum 연산을 줄이기 위해 먼저 기존에 저장한 결과 값에서 이전 값을 제외하고 이후값을 더해 값들을 구하는 방식을 사용하였습니다. 범위 안의 값을 모두 사용하지 않고 단 3개의 값만 계산하면 되므로 기존보다 실행 속도를 대폭 줄일 수 있습니다.

---

Git Hub

BaekJoon-Study/2559.py at python · GUuu9/BaekJoon-Study (github.com)

GitHub - GUuu9/BaekJoon-Study