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

프로그래머스 - 다음에 올 숫자

풀이

1. 무조건 등차수열 아니면 등비수열

2. (첫 번째 요소와 두 번째 요소의 차이) == (두 번째 요소와 세 번째 요소의 차이) ➡️ 등차수열

3. 2가 아니면 ➡️ 등비수열

코드

public class NextNumber {
	public int solution(int[] common) {
		if (Math.abs(common[0] - common[1]) == Math.abs(common[1] - common[2]))
			return (common[1] - common[0]) + common[common.length - 1];
		return (common[1] / common[0]) * common[common.length - 1];
	}

	public static void main(String[] args) {
		NextNumber nextNumber = new NextNumber();
		System.out.println(nextNumber.solution(new int[]{1, 2, 3, 4}));     // 5
		System.out.println(nextNumber.solution(new int[]{2, 4, 8}));        // 16
	}
}