math.comb(n, k)
nCk와 같은 조합 값을 반환
조합은 n개의 수 중 k개를 꺼내는 수와 동일
import math
print("math.comb(5, 1) = " + str(math.comb(5, 1)))
print("math.comb(5, 2) = " + str(math.comb(5, 2)))
print("math.comb(10, 1) = " + str(math.comb(10, 1)))
print("math.comb(10, 4) = " + str(math.comb(10, 4)))
#결과
math.comb(5, 1) = 5
math.comb(5, 2) = 10
math.comb(10, 1) = 10
math.comb(10, 4) = 210
math.perm(n, k)
nPk와 같은 순열 값을 반환
n개의 수 중 k개를 꺼내서 나열하는 경우의 수
import math
print("math.perm(5, 1) = " + str(math.perm(5, 1)))
print("math.perm(5, 2) = " + str(math.perm(5, 2)))
print("math.perm(10, 1) = " + str(math.perm(10, 1)))
print("math.perm(10, 4) = " + str(math.perm(10, 4)))
math.perm(5, 1) = 5
math.perm(5, 2) = 20
math.perm(10, 1) = 10
math.perm(10, 4) = 5040
관련 문제 ) 백준 1010. 다리 놓기
1010번: 다리 놓기
입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트케이스에 대해 강의 서쪽과 동쪽에 있는 사이트의 개수 정수 N, M (0 < N ≤ M < 30)이 주어진다.
www.acmicpc.net
- 파이썬
import math
test=int(input())
for tc in range(test):
n,m=map(int,input().split())
print(math.comb(m,n))
'코딩테스트 > Python 개념' 카테고리의 다른 글
| [Python] 리스트 회전시킬때 deque.rotate() (0) | 2023.04.09 |
|---|---|
| [Python] 리스트 복사 copy.deepcopy( ) (0) | 2023.03.30 |
| [Python] math.factorial (0) | 2023.03.29 |
| [Python] 리스트 값 한번에 출력하기 (0) | 2023.03.25 |
| [Python] 일급 객체 (0) | 2023.03.19 |