JAVA 백준 11050번 이항 계수 1

문제

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

입력

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

출력

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

예제 입력 1 

5 2

예제 출력 1 

10

알고리즘 분류

조합론(combinatorics), 구현(implementation), 수학(math)

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer in;
        in = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(in.nextToken());
        int K = Integer.parseInt(in.nextToken());
        if (K > N / 2) {
            K = N - K;
        }
        System.out.println(bino_coef_dynamic(N, K));
    }
    public static int bino_coef_dynamic(int n, int k) {
        int[][] arr = new int[n + 1][k + 1];

        for (int i = 0; i <= n; i++) {
            for (int j = 0; j <= Math.min(i, k); j++) {
                if (j == 0 || j == i)
                    arr[i][j] = 1;
                else
                    arr[i][j] = arr[i - 1][j] + arr[i - 1][j - 1];
            }
        }
        return arr[n][k];
    }
}