문제설명
등차수열 혹은 등비수열 common이 매개변수로 주어질 때, 마지막 원소 다음으로 올 숫자를 return 하도록 solution 함수를 완성해보세요.
제한사항
- 2 < common의 길이 < 1,000
- -1,000 < common의 원소 < 2,000
- 등차수열 혹은 등비수열이 아닌 경우는 없습니다.
- 공비가 0인 경우는 없습니다.
입출력 예
| common | result |
|---|---|
| [1, 2, 3, 4] | 5 |
| [2, 4, 8] | 16 |
입출력 예 설명
입출력 예 #1
- [1, 2, 3, 4]는 공차가 1인 등차수열이므로 다음에 올 수는 5이다.
입출력 예 #2
- [2, 4, 8]은 공비가 2인 등비수열이므로 다음에 올 수는 16이다.
풀이
등비수열 혹은 등차수열 common의 길이는 최소 3으로 3개 요소를 갖는다.
문제 해결 방식은 등차 수열 여부만 확인을 하고,
등차수열이 아닐경우 등비수열로 판단하여 풀이 하는 방식이다.class Solution {
public int solution(int[] common) {
if ((common[1] - common[0]) == (common[2] - common[1])) {
return common[common.length-1] + (common[2] - common[1]);
} else
return common[common.length-1] * (common[2] / common[1]);
}
}참고
