[백준/파이썬/수학] 11051번 이항 계수 2 코딩테스트 연습풀이 실버1

문제 출처: https://www.acmicpc.net/problem/11051

11051번: 이항 계수 2

 

문제

자연수 N과 정수 K가 주어졌을 때 이항 계수 (NK)를 10,007로 나눈 나머지를 구하는 프로그램을 작성하시오.

 

입력

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

 

출력

 (NK)를 10,007로 나눈 나머지를 출력한다.

 

예제 입력 1

5 2

 

예제 출력 1

10

 

정답

import math
N, K = map(int, input().split())

answer = math.factorial(N) // (math.factorial(K) * math.factorial(N-K))

print(answer%10007)

이항 계수 1과 비슷한 방법을 풀면 되는 문제였습니다.

 

math모듈을 통해 팩토리얼을 가져와서 10007로 나눈 나머지를 출력해줍니다.