프로그래머스 코딩테스트 연습 - 연속된 수의 합 with Kotlin

문제: https://school.programmers.co.kr/learn/courses/30/lessons/120923

프로그래머스

 

문제설명

연속된 세 개의 정수를 더해 12가 되는 경우는 3, 4, 5입니다. 두 정수 num과 total이 주어집니다. 연속된 수 num개를 더한 값이 total이 될 때, 정수 배열을 오름차순으로 담아 return하도록 solution함수를 완성해보세요.

 

제한 사항

• 1 ≤ num ≤ 100
• 0 ≤ total ≤ 1000
• num개의 연속된 수를 더하여 total이 될 수 없는 테스트 케이스는 없습니다.

 

풀이

1. total을 num으로 나눈 중간값 mid 변수를 선언한다.

2. num이 1, 2, 그리고 홀수 짝수일경우로 나눈다.

3. num이 2외의 짝수일 경우 (예를 들어 4) answer은 mid - 1, mid, mid + 1, mid + 2 

4. num이 1외의 홀수일 경우 (예를 들어 3) answer은 mid -1, mid, mid + 1 

class Solution {
    fun solution(num: Int, total: Int): IntArray {
        var answer: IntArray = intArrayOf()
        if(total == 0) {
            for(i in -(num/2) .. (num/2)) {
                answer += i
            }
        } else {
            val mid:Int = total / num
            if(num == 1) {
                answer += total
            } else if (num ==2) {
                for(i in 0 until num) {
                    answer += mid + i
                }
            } else if(num % 2 == 0) { // num이 2 이외의 짝수일 경우
                for(i in -(num/2) + 1  .. (num/2)) {
                    answer += mid + i
                }
            } else { // num 이 1 이외의 홀수일 경우
                for(i in -(num/2) .. (num/2)) {
                    answer += mid + i
                }
            }
        }
        return answer
    }
}

 

다른 사람 풀이

1. (total / num - (num - 1) / 2) => 이 부분은 정확히 이해하지 못했음

2. num이 3, total이 15인 경우, IntArray(num) 을 선언하면 index 값이 0 1 2 으로 들어감

3. index + t => 각각 4 5 6 이 됨 

class Solution {
    fun solution(num: Int, total: Int): IntArray =
    (total / num - (num - 1) / 2).let { t -> IntArray(num) { index -> index + t } }

}

다른 사람의 풀이를 볼 수 있길래 봤음. 코드가 이해가 안되지만 확실히 알 수 있는 점은 나는 코드를 자바처럼 쓰고있다는 점... 조만간 코틀린 책을 사서 코틀린을 코틀린 답게 쓰는 법을 공부할 예정임.