nextnum

1. 문제

2. 입출력

3. 입출력 예시

4. 문제 설계

1. 주어진 숫자가 3개 이고, 모든 입력 케이스는 등차수열 또는 등비수열인 것이 보장 되므로, 세 수 사이의 차가 같으면 등차수열 그렇지 않으면 등비수열로 결정할 수 있음.

5. 전체 코드

//
//  main.swift
//  NextNum
//
//  Created by 이영재 on 2022/03/07.
//MARK: - nextnum

//MARK: - Framework
import Foundation

//MARK: - Function
func solution() -> Void {
    var answer: String = ""
    
    while true {
        guard let input = readLine()?.components(separatedBy: " ") else { return }
        let a1: Int = input.map { Int($0) }[0] ?? 0
        let a2: Int = input.map { Int($0) }[1] ?? 0
        let a3: Int = input.map { Int($0) }[2] ?? 0
        var flag: Bool = false
        var nextNum: Int = 0
        
        if a1 == 0 && a2 == 0 && a3 == 0 {
            break
        }
        
        if (a2 - a1) == (a3 - a2) {
            flag = true
        }
        
        if flag {
            nextNum = a3 + (a2 - a1)
            answer += "AP \(nextNum)\n"
            
        } else {
            nextNum = a3 * (a2 / a1)
            answer += "GP \(nextNum)\n"
        }
    }
    
    print(answer)
}
solution()

 

전체코드는 여기에서 확인할 수 있습니다.