문제
2559번: 수열
첫째 줄에는 두 개의 정수 N과 K가 한 개의 공백을 사이에 두고 순서대로 주어진다. 첫 번째 정수 N은 온도를 측정한 전체 날짜의 수이다. N은 2 이상 100,000 이하이다. 두 번째 정수 K는 합을 구하기
www.acmicpc.net
문제
매일 아침 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
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
Contribute to GUuu9/BaekJoon-Study development by creating an account on GitHub.
github.com
'백준 코드 공부 > 파이썬' 카테고리의 다른 글
백준 11728 : 배열 합치기 (Python) (0) | 2022.02.05 |
---|---|
백준 9655 : 돌 게임 (Python) (0) | 2022.02.04 |
백준 1284 : 집주소 (Python) (0) | 2022.01.24 |
백준 15726 : 이칙연산 (Python) (0) | 2022.01.22 |
백준 1271 : 엄청난 부자 2 (Python) (0) | 2022.01.12 |