본문 바로가기

# 02/Swift - CTP

[Swift] 코딩테스트 연습! Lv0. 구슬을 나누는 경우의 수

반응형
/* 구슬을 나누는 경우의 수

 - 머쓱이는 구슬을 친구들에게 나누어주려고 합니다. 구슬은 모두 다르게 생겼습니다. 머쓱이가 갖고 있는 구슬의 개수 balls와 친구들에게 나누어 줄 구슬 개수 share이 매개변수로 주어질 때, balls개의 구슬 중 share개의 구슬을 고르는 가능한 모든 경우의 수를 return 하는 solution 함수를 완성해주세요.
*/
func solution27(_ balls:Int, _ share:Int) -> Int {
    var mother:Int = 1
    var child:Int = 1
    for i in 1...share {
        child *= (balls+1-i)
        if child%i == 0 {
            child /= i
        } else {
            mother *= i
        }
    }
    return child/mother
}
반응형