[Algorithm] 프로그래머스 - 다음에 올 숫자 (Kotlin)

등차수열 or 등비수열 common이 매개변수로 주어질 때, 마지막 원소 다음으로 올 숫자를 리턴하는 solution()을 완성하라

 

 

common 안의 값들이 등차인지 등비인지 어떻게 구분하는지를 알아채지 못해서 풀지 못했다.

 

https://gyulpiki.tistory.com/entry/%EB%8B%A4%EC%9D%8C%EC%97%90-%EC%98%AC-%EC%88%AB%EC%9E%90-Java-%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%EC%9E%85%EB%AC%B8

다음에 올 숫자 - Java [프로그래머스 입문]

class Solution {
    fun solution(common: IntArray): Int {
        val check1 = common[1] - common[0]
        val check2 = common[2] - common[1]
        var answer = 0
        answer = if (check1 == check2) {
            common[common.size - 1] + check1
        } else {
            common[common.size - 1] * (check2 / check1)
        }
        return answer
    }
}