[백준 4880] 다음수 (Swift)

4880번: 다음수

• 등차수열 : 앞의 수에 일정한 값을 더하는 수열
  • 뒤의 수 - 앞의 수 값이 매번 일정하면 등차수열
• 등비수열 : 앞의 수에 일정한 값을 곱하는 수열
  • 뒤의 수 / 앞의 수 값이 매번 일정하면 등비수열

풀이

while let input = readLine(), input != "0 0 0" {
    let inputArr = input.split(separator: " ").map { Int($0)! }
    if inputArr[2] - inputArr[1] == inputArr[1] - inputArr[0] {
        print("AP \(inputArr[2] + (inputArr[2] - inputArr[1]))")
    } else if inputArr[2] / inputArr[1] == inputArr[1] / inputArr[0] {
        print("GP \(inputArr[2] * (inputArr[2] / inputArr[1]))")
    }
}

저번 문제 풀 때 연습해보려 했던 while문에 let사용하기, 탈출조건 포함하기를 넣어서 풀었다.