프로그래머스 lv1 두 정수 사이의 합

문제를 보자마자 바로 if문을 통해 분기해야겠다는 생각을했다.

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

        return answer.toLong()
    }
}

 

코드는 이렇게 작성했고, 여기서 downTo를 잠시 까먹어서 step -2로 지정하여

에러가 났당 ㅎㄷㄷ

그래서 다시 시도해 보니 

 

다행히도 성공했다

return은 Long타입으로 바뀌어야해서 기존에 Int 타입을 toLong()을 이용하여 형변환했다.