python计算组合数

计算组合数可以使用math库中的comb函数，也可以自己写一个函数来实现。

使用math库中的comb函数：

import math

n = 5
k = 2

result = math.comb(n, k)

print(result) # 输出 10

自己写一个函数实现：

def comb(n, k):
    if k > n:
        return 0
    if k == 0:
        return 1
    numerator = 1
    denominator = 1
    for i in range(k):
        numerator *= n - i
        denominator *= i + 1
    return numerator // denominator

n = 5
k = 2

result = comb(n, k)

print(result) # 输出 10

两个函数的结果都是一样的，都是计算从n个元素中选k个元素的组合数。