[백준 12847] 꿀 아르바이트

문제

윤호는 퍼주기로 유명한 편의점 사장이다. 그의 편의점에는 특이한 임금 체계를 가지고 있다.

• 각 날마다 일의 차이때문에 일마다 급여가 정해져 있다.
• 윤호는  오차 없이 일급을 따박 따박 당일에 준다.
• 정해진 일 수 만큼만 일을 시킨다.
• 한번이라도 퇴직한 자를 다시 취직 시키지 않는다. (만약 취직을 한다면, 일을 시작 한 날부터 끝날 때까지 하루도 빠지면 안 된다.)

무서운 아르바이트를 짤린 준수는 n+1일 후에 001에 월세를 내야 해서 윤호가 사장으로 있는 편의점에 취직하려 한다. 다행히 주변 퇴직자들의 얘기로 급여에 관련해 파악했다. 또한 퇴직자들의 급여 통계를 통해 당장 n일 후까지 일급 정보를 알아냈다. 하지만 준수는 시험을 준비해야 하기에 최대 m일 밖에 일을 할 수 없다.

어제까지 과제를 제출하고 지금도 001에서 자고 있는 준수를 위해 코딩 잘하는 여러분이 일을 해서 벌 수 있는 최대 이익을 준수에게 또 알려주도록 하자.

 

입력

월세를 내기 바로 전 날 까지 인 n (0 < n ≤ 100,000) 일과 일을 할 수 있는 날 m (0 ≤ m ≤ n) 일이 주어진다.

그 다음 줄 에는 1일부터 n일 까지 일급 Ti가 순서대로 주어진다. (0 < Ti ≤ 1,000,000)

 

출력

준수가 일을 해서 벌 수 있는 최대 이익을 출력한다.

---

접근

위의 문제는 특정 구간의 합 중에 최댓값을 구하는 것이다.

따라서 슬라이딩 윈도우를 적용하였다.

초기에 처음부터 M개의 원소를 더한 다음 M+1부터 탐색을 진행한다.

탐색하면서 현재 나오는 숫자를 더하고 이미 더한 숫자 중 처음 값을 빼서 최댓값 비교를 한다.

 

구현

- 초기에 window에 처음부터 M개의 숫자를 더한다.

- 이후 인덱스 M부터 N까지 순차적으로 탐색한다.

- 현재 나오는 숫자를 window에 더하고 i-step을 통해 window에 처음 더한 값을 뺀다.

- 이제 이를 answer와 비교하여 최댓값 갱신을 한다.

import sys
input = sys.stdin.readline

N, M = map(int, input().split())
numbers = list(map(int, input().split()))

step = M
window = sum(numbers[:step])
answer = window
for i in range(step, N):
    window += numbers[i] - numbers[i - step]
    answer = max(answer, window)

print(answer)