[백준/BOJ] 12847번: 꿀 아르바이트

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

12847번: 꿀 아르바이트

문제

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

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

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

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

입력

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

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

출력

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

해설

DP(Dynamic Programming)을 통해 문제를 해결할 수 있다.

하루 임금을 저장할 wage 배열과 dp배열이 있다고 하자.

dp[i] = dp[i] + dp[i-1]

이처럼 하루하루의 임금을 더해나간다.

예를 들어 10 20 30 20 10이 있다 하면,

dp배열에는 10 30 60 80 90이 저장된다.

 

준수는 m일동안의 일을 할 수 있다.

따라서 m일이 3일 때 최소 3번째 인덱스부터 dp배열을 확인해보면 되는 것이다.

var totalWage = 0
for(i in m .. n) {
    if(totalWage <= dp[i] - dp[i - m]) totalWage = dp[i] - dp[i - m]
}

이처럼 dp[i]에서 dp[i - m]의 값을 빼면 m일동안 일했을 때의 총 임금을 알 수 있다.

값이 더 클 시 totalWage를 변경하며 최대 이익을 낼 수 있는 값을 찾으면 된다.

소스 코드

import java.util.*

fun main()  = with(System.`in`.bufferedReader()) {
    val (n, m) = readLine().split(" ").map { it.toInt() }
    val wage = IntArray(n)
    val dp = LongArray(n + 1)
    val st = StringTokenizer(readLine())
    repeat(n) { i ->
        wage[i] = st.nextToken().toInt()
        if(i != 0) dp[i] = wage[i-1].toLong()
    }
    repeat(n) {
        if(it != 0) {
            dp[it] = dp[it] + dp[it - 1]
        }
    }
    var totalWage = 0L
    for(i in m .. n) {
        if(totalWage <= dp[i] - dp[i-m]) totalWage = dp[i] - dp[i-m]
    }
    println(totalWage)
}