백준 2096 내려가기 Kotlin

: ) YOUNG·2023년 8월 27일
1

알고리즘

목록 보기
245/411
post-thumbnail

백준 2096번
https://www.acmicpc.net/problem/2096

문제




생각하기


  • DP 문제이다.

    • 메모이제이션을 활요해서 문제를 풀었다.
    • 최대 메모와 최소 메모를 가지고 답을 구한다.

동작



결과


코드


Kotlin


import java.io.*
import java.util.*
import kotlin.math.max
import kotlin.math.min

// input
private lateinit var br: BufferedReader

// variables
private var N = 0
private lateinit var maxMemo: Array<IntArray>
private lateinit var minMemo: Array<IntArray>

fun main() {
    br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    input()

    bw.write(solve())
    bw.close()
} // End of main()

private fun solve(): String {
    val sb = StringBuilder()

    sb.append(max(maxMemo[N][0], max(maxMemo[N][1], maxMemo[N][2]))).append(' ')
        .append(min(minMemo[N][0], min(minMemo[N][1], minMemo[N][2])))
    return sb.toString()
} // End of solve()

private fun DP(i: Int, first: Int, second: Int, third: Int) {
    // 0 max
    maxMemo[i][0] = maxMemo[i - 1][0] + first
    maxMemo[i][1] = maxMemo[i - 1][0] + second

    // 0 min
    minMemo[i][0] = minMemo[i - 1][0] + first
    minMemo[i][1] = minMemo[i - 1][0] + second

    // 1 max
    // 기존의 값이 최대인지, 새로운 새로값이 최대인지 확인해서 갱신
    maxMemo[i][0] = max(maxMemo[i][0], maxMemo[i - 1][1] + first)
    maxMemo[i][1] = max(maxMemo[i][1], maxMemo[i - 1][1] + second)
    maxMemo[i][2] = maxMemo[i - 1][1] + third

    // 1 min
    minMemo[i][0] = min(minMemo[i][0], minMemo[i - 1][1] + first)
    minMemo[i][1] = min(minMemo[i][1], minMemo[i - 1][1] + second)
    minMemo[i][2] = minMemo[i - 1][1] + third


    // 2 max
    maxMemo[i][1] = max(maxMemo[i][1], maxMemo[i - 1][2] + second)
    maxMemo[i][2] = max(maxMemo[i][2], maxMemo[i - 1][2] + third)

    // 2 min
    minMemo[i][1] = min(minMemo[i][1], minMemo[i - 1][2] + second)
    minMemo[i][2] = min(minMemo[i][2], minMemo[i - 1][2] + third)
} // End of DP()

private fun input() {
    N = br.readLine().toInt()
    maxMemo = Array(N + 1) { IntArray(3) }
    minMemo = Array(N + 1) { IntArray(3) }

    for (i in 1..N) {
        StringTokenizer(br.readLine()).run {
            val first = nextToken().toInt()
            val second = nextToken().toInt()
            val third = nextToken().toInt()

            DP(i, first, second, third)
        }
    }
} // End of input()

0개의 댓글