본문 바로가기
알고리즘/dynamic programming

[백준][Python] 1010번 다리 놓기

by 임짠짠 2022. 5. 5.
반응형
 

1010번: 다리 놓기

입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트케이스에 대해 강의 서쪽과 동쪽에 있는 사이트의 개수 정수 N, M (0 < N ≤ M < 30)이 주어진다.

www.acmicpc.net

 

코드

import sys
import math

n = int(input())
for _ in range(n):
    a, b = map(int,sys.stdin.readline().split())
    ans = math.factorial(b)//(math.factorial(b-a)*math.factorial(a))
    print(ans)

 

설명

a와 b를 입력받으면 b가 a보다 항상 크거나 같기 때문에 b개 중에 a개를 뽑는 경우의 수를 구하면 된다. 다리끼리 겹치면 안되므로 순서는 바뀔 수 없다.

서로 다른 n개에서 r개를 선택하는 경우의 수

위 식을 이용해서 경우의 수를 구하면 된다. 

팩토리얼 계산은 math 라이브러리에서 제공하는 factorial 함수를 사용했다.

반응형

댓글