본문 바로가기

알고리즘

[백준][KOTLIN] 2579 계단 오르기

728x90

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

 

2579번: 계단 오르기

계단 오르기 게임은 계단 아래 시작점부터 계단 꼭대기에 위치한 도착점까지 가는 게임이다. <그림 1>과 같이 각각의 계단에는 일정한 점수가 쓰여 있는데 계단을 밟으면 그 계단에 쓰여 있는 점

www.acmicpc.net

풀이

코드

import java.util.*

fun main(args: Array<String>) {
    val sc = Scanner(System.`in`)
    val n = sc.nextInt()
    val score = IntArray(301)

    for (i in 1..n) {
        score[i] = sc.nextInt()
    }

    val dp = IntArray(301)
    dp[1] = score[1]
    dp[2] = score[2] + dp[1]

    for (i in 3..n) {
        dp[i] = score[i] + max(score[i - 1] + dp[i - 3], dp[i - 2])
    }

    println("${dp[n]}")
}

private fun max(a: Int, b: Int): Int {
    return if (a < b) b else a
}
728x90