(Swift) 백준 10952 A + B - 5, 10951 A + B - 4 - 갯수가 정해지지 않은 input 받기

SteadySlower·2022년 5월 7일
0

Coding Test

목록 보기
14/298

마지막 테스트 케이스의 조건이 주어질 때

10952번 - A+B - 5

마지막 테스트 케이스의 조건이 주어진 경우에는 일단 while true를 통해서 계속 input을 받도록 하고 중간에 마지막 input의 조건을 만족하는 경우 break문을 사용하면 됩니다.

import Foundation

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

while true {
    let inputs = readLine()!.components(separatedBy: " ").map { Int($0)! }
    let a = inputs[0]
    let b = inputs[1]
    let sum = a + b
    if sum == 0 {
        break
    }
    print(a + b)
}

마지막 테스트 케이스의 조건이 주어지지 않을 때

10951번 - A+B - 4

마지막 테스트 케이스의 조건이 주어지지 않을 때는 readLine이 input이 더 이상 없을 때 nil을 return한다는 점을 활용하면 됩니다.

import Foundation

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

while let inputs = readLine() {
    let inputArray = inputs.components(separatedBy: " ").map{ Int($0)! }
    let a = inputArray[0]
    let b = inputArray[1]
    print(a + b)
}
profile
백과사전 보다 항해일지(혹은 표류일지)를 지향합니다.

0개의 댓글