프로그래머스 LV.0 다음에 올 숫자

프로그래머스 LV.0 다음에 올 숫자

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

프로그래머스

 

1. 문제 명

다음에 올 숫자

2. 문제 설명

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

3. 제한 사항

2 < common의 길이 < 1,000

-1,000 < common의 원소 < 2,000

common의 원소는 모두 정수입니다.

등차수열 혹은 등비수열이 아닌 경우는 없습니다.

등비수열인 경우 공비는 0이 아닌 정수입니다.

4. 예시

common	result
[1, 2, 3, 4]	5
[2, 4, 8]	16

5. 기본 제공 코드

function solution(common) {
    var answer = 0;
    return answer;
}

6. 제출한 내 답

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

 

6-2. VSC에 작성한 내용

const solution = (common) => {
  /** 등차수열 혹은 등비수열이 주어진다.
   * 다음에 올 수를 반환하라
   */

  // 아래와 같이 해보고, 안되면 common.reverse한 다음 테스트
  if (common[1] - common[0] === common[2] - common[1]) {
    // 등차수열의 경우
    return common[common.length - 1] + (common[1] - common[0]);
  } else if (common[1] / common[0] === common[2] / common[1]) {
    // 등비수열의 경우
    return common[common.length - 1] * (common[1] / common[0]);
  }
};

// test
console.log(solution([1, 2, 3, 4]));
console.log(solution([2, 4, 8]));

7. 특이사항

없음

8. 다른 사람이 작성한 답

8-1. pop

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

배열[배열.length - 1]은 배열.pop이라고도 표현할 수 있다.

이건 자주 쓸 것 같으니 기억해두자.