[Programmers] 120924번 - 다음에 올 숫자

Problem

문제 보기

등차수열 혹은 등비수열 common이 매개변수로 주어질 때, 마지막 원소 다음으로 올 숫자를 return 하도록 solution 함수를 완성해보세요.

---

Solution

function solution(common) {
    if ((common[1] / common[0]) === (common[2] / common[1])) return common[common.length-1] * (common[1] / common[0]);
    else return common[common.length-1] + (common[1] - common[0]);
}

등차수열 : 2번째 요소 - 1번째 요소
등비수열 : 2번째 요소 / 1번째 요소
최소 3개의 원소가 들어간다고 되어있어서 3개를 비교하여 등차수열인지 등비수열인지 비교하는 코드를 작성하였다.