[백준 2559번] 수열

문제 보기

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

2559번: 수열

 

 

소스 코드

import sys
input = sys.stdin.readline

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

num = 0 # 부분합
cnt = 0 # 합한 원소의 개수

answer = []
for i in range(len(arr)):
    num += arr[i]
    cnt += 1
    if cnt == k:
        answer.append(num)
        cnt -= 1
        num -= arr[i-k+1]

print(max(answer))

 

문제 해결

사실 문제의 그림에서 힌트가 있었다. 

리스트에서 반복문을 돌려준 다음, 합한 원소의 개수가 k개가 되었을 때 그 합을 저장해준다.

그 후, 리스트의 가장 앞에 있는 원소 하나만 빼주고 다시 다음 원소를 더하는 과정을 반복해서 해결한다.