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

 

📚 문제 설명

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

 

✅ 제한사항

• 2 < common의 길이 < 1,000
• 1,000 < common의 원소 < 2,000
  • common의 원소는 모두 정수입니다.
• 등차수열 혹은 등비수열이 아닌 경우는 없습니다.
• 등비수열인 경우 공비는 0이 아닌 정수입니다.

 

💻 구현 코드

public int solution(int[] common) {

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

}

 

 

📌 구현코드 해설

common 가 최소 3이기 때문에 0, 1, 2 를 사용하여 1-0 , 2-1 의 차이가 같다면 등차일 것이고

그 외인 경우는 등비일 것이기 때문에 if- else문을 사용하였다. 

또한 data를 선언하여 공차, 공비를 구한후 return할때 사용한다. 

 

 

🍏 회고 / 새롭게 알게 된 점

 

문제에서 2가지 경우를 사용한다면 하나의 경우는 if에 두고 그 외는 else를 사용하여 간단하게 풀 수 있다는 사실을 알게되었다.