[백준] 1731 추론

https://www.acmicpc.net/problem/1731

1731번: 추론

 

이 문제는 입력값들로부터 간단한 규칙성을 찾은 후 결과값을 출력하는 문제였다. 어차피 규칙은 등차나 등비 둘 중 무조건 한 가지였기 때문에 케이스를 두 개로 나눠서 간단하게 처리할 수 있었다.

 

---

문제 및 입출력

---

예제

---

코드

import java.util.Scanner;

public class Main
{
    public static void main(String[] args)
    {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] arr = new int[n];
        
        for(int i=0; i<n; i++)
        {
            arr[i] = sc.nextInt();
        }
        
        int result = arr[arr.length-1];
        
        if(arr[2]-arr[1]==arr[1]-arr[0])
        {
            System.out.println(result+arr[arr.length-2]-arr[arr.length-3]);
            return;
        }
        else
        {
            System.out.println(result*(arr[arr.length-2]/arr[arr.length-3]));
            return;
        }
    }
}