[백준] 추론 (1731)(kotlin)

문제 설명

백준 1731번 문제 링크

입력 및 출력

» 입력

첫째 줄에 수열의 길이 N이 주어진다. 둘째 줄부터 N개의 줄에 수열의 각 원소가 차례대로 주어진다. 주어지는 수열은 등차수열이나 등비수열 중에 하나다. N은 항상 3 이상 50이하이며, 입력되는 수는 10^6 이하의 자연수이다.

» 출력

첫째 줄에 수열의 다음 원소를 출력한다. 이 수는 20억보다 작거나 같다.

예제 입출력(테스트케이스)

입력	출력
4 364 843 1322 1801	2280

문제 풀이1

fun main(args: Array<String>) = with(System.`in`.bufferedReader()) {
    val N = readLine().toInt()
    val arr = IntArray(N) { readLine().toInt() }

    println(
        if (arr[1] - arr[0] == arr[2] - arr[1]) {
            arr[arr.size - 1] + arr[1] - arr[0]
        } else {
            arr[arr.size - 1] * (arr[1] / arr[0])
        }
    )
}