[Kotlin] 두 정수 사이의 합 (프로그래머스 알고리즘)

 

문제

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

 

 

---

 

접근

 

• 무의식중에 당연히 입력되는 값 a, b에 대해서 a < b라 생각했고
  그냥 a부터 b까지 for 문을 돌리고 sum이라는 변수에 계속 더해서 return하면 된다고 생각했다.

// 구체적인 구상, 오답 코드

class Solution {
    fun solution(a: Int, b: Int): Long {
        var sumInt = 0
        for(i in a .. b){
            var sumInt = sumInt + i
        }
        return sumInt.toLong()
    }
}

• 그런데 위와 같이 코드를 작성했더니 결과값이 모두 0이 나왔다.
• 문제를 다시보니 a와 b중 큰 값이 뭔지 모르고, 해당 값에 따라 분기해야 하는 문제였다.

// 테스트 케이스 1, 3 실패

class Solution {
    fun solution(a: Int, b: Int): Long {
        var sumInt = 0
        if(a < b){
            for(i in a .. b){
                var sumInt = sumInt + i
            }
        }
        else if(a > b){
            for(i in b .. a){
                var sumInt = sumInt + i
            }
        } else sumInt = a
        return sumInt.toLong()
    }
}

• 그래서 위와 같이 구성했는데 또 틀렸다.
• 이유는 for문 안에서 sumInt 앞에 var을 계속 선언했기 때문이다.
  위에서 선언해놓고 안에서 다시 별개의 지역변수를 선언한 셈이다.
• for문 안에 있는 var를 삭제해주니 정상적으로 작동되었다.

// 테스트 케이스 4~10 실패

class Solution {
    fun solution(a: Int, b: Int): Long {
        var sumInt = 0
        if(a < b){
            for(i in a .. b){
                sumInt = sumInt + i
            }
        }
        else if(a > b){
            for(i in b .. a){
                sumInt = sumInt + i
            }
        } else sumInt = a
        return sumInt.toLong()
    }
}

• 그런데 이번엔 테스트케이스에서 실패가 몇 개 떴다.
• 아무래도 문제에서 반환형을 Long으로 설정한 거 보면 Int를 넘어서는 범위의 수를 출력하는 것 같았다.
  sumInt를 다 계산해놓고 뒤늦게 toLong()을 해봤자 의미가 없는 거다.
• 모든 선언형을 미리 Long으로 변환해보았더니 문제가 해결 됐다. 코드도 좀 더러워보여서 수정했다.

 

---

 

성공 코드

 

class Solution {
    fun solution(a: Int, b: Int): Long {
        var sumLong: Long = 0 // Long으로 선언
        
        val start = if (a < b) a else b
        val end = if (a < b) b else a
        
        for (i in start..end) {
            sumLong += i // Long으로 누적
        }
        
        return sumLong
    }
}

 

---

 

설명

 

1. 모든 값을 저장할 변수 sumLong 를 Long 형태로 선언한다. (: Long 으로 표기하든지, 0L로 선언)
2. a와 b의 크기에 따라 start와 end 변수에 각각 집어 넣는다.
3. start부터 end까지 for문을 돌리면서 더한다.
   여기서 sumLong을 미리 Long으로 선언했으므로 Long으로 계속 더해진다.
   (까먹고 예외처리를 못했는데, a와 b가 같은 경우는 if~else 문으로 분기하면 된다.)
4. 값이 모두 더해진 sumLong을 리턴한다.

 

---

 

다른 방법

 

class Solution {
    fun solution(a: Int, b: Int) =
        if (a < b) ((a..b).average() * (b - a + 1)).toLong()
        else if (a > b) ((b..a).average() * (a - b + 1)).toLong()
        else a.toLong()
}

class Solution {
    fun solution(a: Int, b: Int): Long {
        var answer = 0L

        if (a == b){
            return a.toLong()
        }

        var min = Integer.min(a,b)
        var max = Integer.max(a,b)

        for (i in min .. max){
            answer += i
        }

        return answer
    }
}

• average()나 min, max()를 이용해서 푸신 분들이 있었다.