[백준 6322] 직각 삼각형의 두 변 (Swift)

6322번: 직각 삼각형의 두 변

• a제곱 + b제곱 = c제곱 을 만족하는 양수 a, b, c 구하기
  • abc 중 2개는 자연수로 주어짐
  • 미지수가 a라면
    • a = 제곱근(c제곱 - b제곱)
  • 미지수가 b라면
    • b = 제곱근(c제곱 - a제곱)
  • 미지수가 c라면
    • c = 제곱근(a제곱 + b제곱)
• 제곱근 안의 수가 0보다 “작거나”(음수), 0일 때 직각삼각형을 만들 수 없다.
  • 음수 조건 뿐 아니라 0이 되는 조건도 따져야 함에 유의
• Swift의 sqrt() 함수는 파라미터가 음수일 때 nan을 반환한다.
  • nan은 type(of:) 로 판별했을 땐 Double이지만, == 로 판별할 수 없다.
  • 그래서 일단 sqrt() 내부에 넣을 수가 0보다 큰지 판별해서 풀었다.
  • 공식문서를 찾아보니 Double.nan 과 같이 사용 가능하고, == 대신 isNan 으로 판별하라고 한다!
  • https://developer.apple.com/documentation/swift/floatingpoint/nan
• 출력 형식에서 빈 줄 출력에 유의

풀이

import Foundation
var count = 0
while true {
    let input = readLine()!
    if input == "0 0 0" {
        break
    }
    if count > 0 {
        print("")
    }
    let inputArr = input.split(separator: " ").map { Int($0)! }
    let a = inputArr[0]
    let b = inputArr[1]
    let c = inputArr[2]
    if a < 0 {
        let answer = Double(c*c - b*b)
        print("Triangle #\(count + 1)")
        if answer <= 0 {
            print("Impossible.")
        } else {
            print("a = \(String(format: "%.3f", sqrt(answer)))")
        }
    }
    if b < 0 {
        let answer = Double(c*c - a*a)
        print("Triangle #\(count + 1)")
        if answer <= 0 {
            print("Impossible.")
        } else {
            print("b = \(String(format: "%.3f", sqrt(answer)))")
        }
    }
    if c < 0 {
        let answer = Double(a*a + b*b)
        print("Triangle #\(count + 1)")
        if answer <= 0 {
            print("Impossible.")
        } else {
            print("c = \(String(format: "%.3f", sqrt(answer)))")
        }
    }
    count += 1
}