[알고리즘 문제/2020 KAKAO BLIND RECRUITMENT] 문자열 압축

https://programmers.co.kr/learn/courses/30/lessons/60057

코딩테스트 연습 - 문자열 압축

 

이슈

제출 테스트 5번만 실패

 

 

---

 

해결

import kotlin.test.*

class Solution {

    fun solution(s : String) : Int {

        var result = Int.MAX_VALUE
        if(s.length == 1) result = 1
        for(unitCount in 1 .. (s.count() - 1)){
            val zipResult = zipString(unitCount, s)
            if(result > zipResult) result = zipResult
        }

        return result
    }

    fun zipString(unitCount: Int, s: String) : Int {    
        var result = 0
        var zipCount = 1
        var prevUnit = ""
        for (i in 0..(s.count()) step unitCount) {
            var currentUnit = getUnit(i, unitCount, s)  
            if(prevUnit == currentUnit) {
                zipCount++
            } else {
                if(zipCount > 1){
                    result += (zipCount.toString().length + currentUnit.length)
                } else {
                    result += currentUnit.length
                }
                zipCount = 1
                prevUnit = currentUnit
            }
        }
        return result
    }
    
    fun getUnit(index: Int, unitCount: Int, s: String) : String {
        return if(index + unitCount > s.count()) {
            return s.subSequence(index, s.count()).toString()
        } else {
            s.subSequence(index, index + unitCount).toString()
        }       
    }
}

class SolutionTest {

    val solution = Solution()


    fun testGetUnit() {

        val testCase1 = Pair(Triple(3, 3, "ababcdcdababcdcd"), "bcd")
        val testCase2 = Pair(Triple(0, 4, "ababcdcdababcdcd"), "abab")
        val testCase3 = Pair(Triple(8, 8, "ababcdcdababcdcd"), "ababcdcd")
        val testCase4 = Pair(Triple(6, 6, "abcabcdede"), "dede")
        val testCase5 = Pair(Triple(0, 3, "abcabcdede"), "abc")

		val testCases = listOf(testCase1 , testCase2, testCase3, testCase4, testCase5)

		for (test in testCases) {
            val (quest, answer) = test
            val (index, unitCount, s) = quest
            val result = solution.getUnit(index, unitCount, s.trim())
            assertEquals(answer, result, "epected : $answer, actural : $result")
        }
    }
    
    

    fun testZipString() {

        val testCase1 = Triple("ababcdcdababcdcd", 1, "ababcdcdababcdcd")
        val testCase2 = Triple("ababcdcdababcdcd", 2, "2ab2cd2ab2cd")
        val testCase3 = Triple("ababcdcdababcdcd",8, "2ababcdcd")
        val testCase4 = Triple("abcabcdede", 2, "abcabc2de")
        val testCase5 = Triple("abcabcdede",3, "2abcdede")

		val testCases = listOf(testCase1 , testCase2, testCase3, testCase4, testCase5)

		for (test in testCases) {
            val (s, unitCount, answer) = test
            val result = solution.zipString(unitCount, s)
            assertEquals(answer.length, result, "epected : ${answer.length}, actural : $result")
        }
    }


    fun testSolution() {

        val testCase1 = Pair("aabbaccc", 7)
        val testCase2 = Pair("ababcdcdababcdcd",9)
        val testCase3 = Pair("abcabcdede",8)
        val testCase4 = Pair("abcabcabcabcdededededede",14)
        val testCase5 = Pair("xababcdcdababcdcd",17)

        val testCases = listOf(testCase1 , testCase2, testCase3, testCase4, testCase5)

        for(test in testCases) {
            val (string, answer) = test
            val result = solution.solution(string)
            assertEquals(answer, result, "epected : $answer, actural : $result")
        } 
    }

}

fun main() {    
    val solutionTest = SolutionTest() 

	solutionTest.testGetUnit()
    solutionTest.testZipString()
    solutionTest.testSolution()

}

5번문제는 입력 스트링이 1글자인 경우 테스트로 추측

8번째줄의 한글자인 경우 예외처리로 테스트에 통과할 수 있었다