[swift] 프로그래머스 - 구슬을 나누는 경우의 수

프로그래머스

---

문제풀이를 직관적으로 이해할 수 있도록 고차함수를 사용하지 않은 풀이와
고차함수를 사용한 풀이로 나누어 공부하고 있습니다.

 

처음 제출했던 풀이(오답 주의!)

func solution(_ balls:Int, _ share:Int) -> Int {
    func fac(_ n: Int) -> Int {
        return (1...n).reduce(1,*)
    }
    return fac(balls) / fac(balls-share) * fac(share)
}

• 팩토리얼 함수를 만들어서 공식대로 대입했지만
• 꽤 많은 문제에서 오류가 나왔다.
• 이유는 계산중에 값이 넘쳐서 그런 거 같다. (정확하지 않음)
• 여러 풀이 사례를 보고 내 식대로 고쳐봤다.

 

수정해서 통과한 풀이

import Foundation

func solution(_ balls:Int, _ share:Int) -> Int {
    guard balls != share else { return 1 }
    func fac(_ n: Int) -> Double {
        return Double((1...n).reduce(1.0){ Double($0) * Double($1) })
    }
    
    return Int(round(fac(balls) / fac(share) / fac(balls-share)))
}

• balls와 share의 수가 같을 때는 경우의 수가 1이니 guard문으로 return을 빼줬다.
• 팩토리얼 함수를 싹 Double로 바꿔줬다. Double이 표현 범위가 더 큰가..? 공부 좀 해봐야겠다.

Int와 Double의 최대 표현 값

• return 값을 Int로 바꾸기전에 round로 소수점을 반올림했다.
• n! / (n-m)! * m! 공식을 n! / m! / (n-m)!로 바꿔줬다.
• 이렇게 바꾸니 문제를 모두 통과했지만 근본적인 문제해결을 위해 좀 더 접근해봐야 할 듯하다.
• core dump 에러 너무 싫다.