백준 22.03.20. 11050번 - 이항 계수 1

이항 정리 - 위키백과, 우리 모두의 백과사전

이항정리, 이항계수를 알아봅시다.

• 수학시간에 방정식이라는 것을 배워본 적이 있을겁니다.
• 그 중 두 항을 더하는 식의 제곱을 구하는 공식은 (a+b)² = a² + 2ab + b² 이며
• 두 항을 더하는 식의 세제곱을 구하는 공식은 (a+b)³ = a³ + 3a²b + 3ab² + b³ 입니다.
• 하지만 (a+b)ⁿ = xaⁿ⁻¹b + yaⁿ⁻²b² + … + bⁿ 처럼 n제곱을 구하는 식에서의 결괏값의 각 항의 계수는 구하는 공식은 잘 모릅니다.
• 이런 이항식의 결괏값들이 가지는 계수를 계산할 수 있도록 도와주는 식이 이항정리이며 이항계수는 위 식에서 x, y와 같이 각 단항식들이 가지는 계수입니다.
• 결론을 말하자면 (a+b)ⁿ의 결과로 나오는 단항식들이 가지는 이항계수는 조합의 형태인 nCk의 값을 가집니다.
  • (a+b)ⁿ = nC₁aⁿ⁻¹b + nC₂aⁿ⁻²b² + … + bⁿ 처럼 계산할 수 있습니다.
  • nCk = n! / (n-k)! k! 의형태를 가집니다.

---

from math import factorial


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

print(factorial(N)//(factorial(K)*factorial(N-K)))

• 받은 두 수를 바탕으로 조합식을 짜주면 됩니다.
• 이번 코드같은 경우는 factorial모듈을 임포트하여 사용하였습니다.