[백준] 2407(파이썬) - 조합

resilient

·

2021. 8. 12. 14:25

728x90
반응형

https://www.acmicpc.net/problem/2407

 

2407번: 조합

n과 m이 주어진다. (5 ≤ n ≤ 100, 5 ≤ m ≤ 100, m ≤ n)

www.acmicpc.net

  • 이 문제는 조합 공식을 그대로 파이썬으로 구현하면 되는 문제지만 다른 방법이 뭐가 있을까 하다가 아래와 같이 구현했다.
  • math.factorial을 사용해서 조합 공식을 그대로 적용해도 같은 답이 나온다.
import sys
input = sys.stdin.readline

n,m = map(int,input().split())

data = [i for i in range(1,n+1)]

ans1 = 1
ans2 = 1
for i in data[-m:]:
    ans1 *= i
for i in data[:m]:
    ans2 *= i

print(ans1//ans2)
반응형