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

 

Level. 0

 

문제

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

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

 

풀이

class Solution {
    public int solution(int[] common) {
        int answer = 0;
        int a = common[1] - common[0];
        int b = common[2] - common[1];
        if(a==b) answer = common[common.length-1] + a; // 등차수열
        else  answer = common[common.length-1] * (common[2] / common[1]); // 등비수열
        return answer;
    }
}

 

 

---

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

프로그래머스