9251번: LCS - Kotlin

9251번: LCS

문제

LCS(Longest Common Subsequence, 최장 공통 부분 수열)문제는 두 수열이 주어졌을 때, 모두의 부분 수열이 되는 수열 중 가장 긴 것을 찾는 문제이다.

예를 들어, ACAYKP와 CAPCAK의 LCS는 ACAK가 된다.

입력

첫째 줄과 둘째 줄에 두 문자열이 주어진다. 문자열은 알파벳 대문자로만 이루어져 있으며, 최대 1000글자로 이루어져 있다.

출력

첫째 줄에 입력으로 주어진 두 문자열의 LCS의 길이를 출력한다.

 

나의 풀이

 이 문제는 DP 문제이다. 이전에 평범한 배낭 문제로 DP를 접하였기 때문에 이전에 풀었던 방식을 응용하여 문제를 해결하였다.

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

fun main() = with(System.`in`.bufferedReader()) {
    val sequenceA = readLine().toCharArray()
    val sequenceB = readLine().toCharArray()

    val sizeA = sequenceA.size
    val sizeB = sequenceB.size
    val dp = Array(sizeA + 1) { IntArray(sizeB + 1) }

    for (i in 1 until sizeA + 1) {
        for (j in 1 until sizeB + 1) {
            if (sequenceA[i - 1] == sequenceB[j - 1]) {
                dp[i][j] = dp[i - 1][j - 1] + 1
            }
            else {
                dp[i][j] = max(dp[i][j - 1], dp[i - 1][j])
            }
        }
    }
    println(dp[sizeA][sizeB])
}