[백준 9094] 수학적 호기심 (Swift)

9094번: 수학적 호기심

• 범위가 크지 않고(0…100) 정수형이므로 모든 경우의 수를 계산했다.
• Int인 Double 판별, 즉 Double의 소수점 이하가 0인지 아닌지 판별하기는 여기에 따로 정리해 두었다.

풀이

(69104KB, 108ms)

let count = Int(readLine()!)!
(1...count).forEach { _ in
    let input = readLine()!.split(separator: " ").map { Int($0)! }
    let n = input[0]
    let m = input[1]
    var pairs = 0
    (1..<n).forEach { a in
        (a+1..<n).forEach { b in // a < b
            let value = Double(((a*a) + (b*b) + m)) / Double(a*b) // Int끼리 나누면 소수점을 버림
            if value == Double(Int(value)) { // Double이 Int인지 아닌지 판별
                pairs += 1
            }
        }
    }
    print(pairs)
}

풀이2

다른 풀이들을 보고 (a제곱) + (b제곱) + m 을 (a*b) 로 나눈 나머지 (% 사용)가 0이면 소수점 이하가 0임을 판별할 수 있어서 그렇게도 해봤는데, 이쪽이 시간이 조금 더 걸렸다.

(69104KB, 128ms)

let count = Int(readLine()!)!
(1...count).forEach { _ in
    let input = readLine()!.split(separator: " ").map { Int($0)! }
    let n = input[0]
    let m = input[1]
    var pairs = 0
    (1..<n).forEach { a in
        (a+1..<n).forEach { b in
            let value = (a*a) + (b*b) + m
            if value % (a*b) == 0 {
                pairs += 1
            }
        }
    }
    print(pairs)
}