[Kotlin] 수박수박수박수박수박수? (프로그래머스 알고리즘)

 

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12922

 

 

 

---

 

성공 코드

 

class Solution {
    fun solution(n: Int): String {
        
        var returnArray = CharArray(n)
        var i = 0
        while(i < n){
                if(i % 2 == 0) returnArray[i] = '수'
                else returnArray[i] = '박'
                i++ }
        
            return returnArray.joinToString("")
    }
}

 

---

 

설명

 

1. var returnArray = CharArray(n)
   숫자 n 크기만큼의 정답 배열을 CharArray 형태로 선언해둔다.
2. var i = 0; while(i < n){ if(i % 2 == 0) returnArray[i] = '수' ... i++ }
   while문 안의 if문을 n번째 반복하며, 짝수/홀수에 따라 정답배열에 '수박수박'을 채운다.
3. return returnArray.joinToString("")
   결과배열을 구분자 없이 String으로 리턴한다.
4. 아쉽게도 이 방법은 매우 느리다. 보다 효율적인 코드들은 아래에 있다.

 

---

 

다른 방법

 

class Solution {
    fun solution(n: Int): String = String(CharArray(n,{i-> if(i%2==0) '수' else '박'}))
}

• 실행에 1초가 채 걸리지 않는 풀이다. 람다 형식을 사용했다.
• CharArray(n, { i -> if (i % 2 == 0) '수' else '박' })
  길이가 n인 문자 배열을 생성한다. 배열의 각 요소는 인덱스 i의 짝/홀수임에 따라 '수' 또는 '박' 문자로 초기화된다. 이를 위해 코틀린의 CharArray 생성자를 사용하며, 람다 식을 전달하여 각 요소를 초기화한다.
• { i -> if (i % 2 == 0) '수' else '박' }
  이 표현식은 i를 매개변수로 받아서 if 조건에 따라 '수' 또는 '박' 문자를 반환한다. 람다 표현식은 코드를 간결하게 만들어주고, 함수를 인라인으로 전달할 수 있는 장점을 제공한다. 이 코드에서는 CharArray 생성자에 익명 함수를 전달하여 배열 요소를 초기화하는 것을 간결하게 표현한 것이다.
• String(...)
  String 생성자를 사용하여 문자 배열을 기반으로 새로운 문자열로 변환한다.

 

class Solution {
    fun solution(n: Int): String {
        return "수박".repeat(n / 2) + if (n % 2 != 0) "수" else ""
    }
}

• "수박".repeat(n / 2)
  문자열 "수박"을 n / 2번 반복한 결과를 반환한다.
  예를 들어, n이 3이면 "수박"이 반환되고, n이 4이면 "수박수박"이 반환된다.
• if (n % 2 != 0) "수" else ""
  만약 n이 홀수인 경우, "수"를 반환하고, 짝수인 경우에는 빈 문자열을 반환한다.
  이를 통해 문자열의 길이가 홀수인 경우 마지막에 "수"를 추가하여 패턴을 완성한다.
• else "" 가 무조건 필요한 이유?
  주어진 코드에서 else 부분이 없으면, if 조건식이 참일 때만 "수"가 반환되고, 거짓일 때는 아무런 값도 반환하지 않게  된다. 그런데 Kotlin의 함수는 반드시 값을 반환해야 한다. 따라서 else 부분이 없으면 반환값이 없는 상황이 생겨 컴파일 오류가 발생한다. 만약 if식이 항상 참이어서 else가 필요 없는 경우엔 생략 가능하지만, 이 경우는 그게 아니므로 무조건 else부분을 명시해주어야 한다.