BOJ(11050) - Kotlin

문제

자연수 \(N\)과 정수 \(K\)가 주어졌을 때 이항 계수 \(\binom{N}{K}\)를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 \(N\)과 \(K\)가 주어진다. (1 ≤ \(N\) ≤ 10, 0 ≤ \(K\) ≤ \(N\))

출력

 \(\binom{N}{K}\)를 출력한다.

 

나의 접근방법

이항계수가 아예 생각나지 않아서 구글링을 했다. 이항계수의 공식은 \(\frac{n!}{r!(n-r)!}\)이였다. 공식에서 팩토리얼을 많이 사용하기 때문에 팩토리얼 함수를 만들고 공식을 그대로 대입시켜 풀었다.

코드

/* 이항 계수 1 */

import java.util.Scanner

fun main() = with(Scanner(System.`in`)) {
    val n = nextInt()
    val r = nextInt()

    print(factorial(n) / (factorial(r) * factorial(n-r)))
}

fun factorial(x: Int): Int {
    if (x <= 1) {
        return 1
    }
    return x * factorial(x-1)
}