[백준 1731] 추론

1731번: 추론

접근

• 등차인지 등비인지 먼저 구해야 한다
  • (2번째 요소 - 1번째 요소) 의 값 == (3번째 요소 - 2번째 요소) 의 값 이면 등차수열
  • (2번째 요소 / 1번째 요소) 의 값 == (3번째 요소 / 2번째 요소) 의 값 이면 등비수열
• 맨 끝 요소의 다음 수를 구해야 하므로 입력을 끝까지 받아야 한다
• 등차수열이라면 맨 끝 요소에 등차를 더한 값을 출력 후 리턴
• 등비수열이라면 맨 끝 요소에 등비를 곱한 값을 출력 후 리턴

func q1731() { // 추론
    let inputCount = Int(readLine()!)!
    var inputArr: [Int] = []
    for _ in 1...inputCount {
        let input = Int(readLine()!)!
        inputArr.append(input)
    }
    let plus = inputArr[1] - inputArr[0]
    if plus == (inputArr[2] - inputArr[1]) {
        print(inputArr[inputCount - 1] + plus)
        return
    }
    let multiple = inputArr[1] / inputArr[0]
    if multiple == (inputArr[2] / inputArr[1]) {
        print(inputArr[inputCount - 1] * multiple)
        return
    }
}