[백준 1731 JAVA 자바] 추론

n개의 배열이 나오는데

이 배열이

등비수열인지 등차수열인지 판단해서

그 다음에 올 숫자를 출력하는 문제입니

 

코드입니다

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        
        int n = scan.nextInt();
        
        int[] arr = new int[n];
        
        for(int i=0; i<n; i++) {
        	arr[i] = scan.nextInt();
        }
        
        int gap = arr[1]-arr[0];
        int div = arr[1]/arr[0];
        
        if(arr[n-1]-arr[n-2]==gap) {
        	System.out.print(arr[n-1]+gap);
        }else {
        	System.out.print(arr[n-1]*div);
        }
    }
}