ps:problems:boj:2225
합분해
| ps | |
|---|---|
| 링크 | acmicpc.net/… |
| 출처 | BOJ |
| 문제 번호 | 2225 |
| 문제명 | 합분해 |
| 레벨 | 골드 5 |
| 분류 |
수학 |
| 시간복잡도 | O(n+k) |
| 인풋사이즈 | n<=200, k<=200 |
| 사용한 언어 | Python |
| 제출기록 | 31312KB / 108ms |
| 최고기록 | 56ms |
| 해결날짜 | 2021/12/31 |
풀이
- 문제를 바꿔쓰면, 학교에서 경우의 수를 배울때 풀던 기본 문제인 'N개의 사과를 K명에게 나눠주는 방법의 수를 구하시오'와 같은 문제이다. 이 문제의 답은 중복조합 ${_{N}H_{K}} = {_{N+K-1}C_{N}} $ 으로 표현된다.
- 모듈러가 소수가 아니라서 효과적으로 구하려면 이항 계수 (Binomial Coefficient)를 구하는 방법중에서도 복잡한 방법을 사용해야 하지만, N과 K가 작기때문에 그냥 파스칼의 삼각형을 이용한 O((N+K)^2) 방법을 써도 되고, 파이썬에서는 그냥 자체 biginteger을 믿고 정확한 값을 계산한 다음에 마지막에 나머지를 취해줘도 된다.
코드
"""Solution code for "BOJ 2225. 합분해".
- Problem link: https://www.acmicpc.net/problem/2225
- Solution link: http://www.teferi.net/ps/problems/boj/2225
Tags: [Math] [Combinatorics]
"""
import math
MOD = 1_000_000_000
def main():
N, K = [int(x) for x in input().split()]
print(math.comb(N + K - 1, N) % MOD)
if __name__ == '__main__':
main()
ps/problems/boj/2225.txt · 마지막으로 수정됨: 2021/12/31 17:15 저자 teferi

토론