Python编写组合数C(n,k)计算函数持续报错问题求助

原代码核心错误点 #

• 组合数基础公式记忆错误：组合数定义为 C(n,k) = n! / (k! * (n-k)!)，合法输入要求 n ≥ k ≥ 0。原代码中第三个阶乘的计算项写为c = k - n，顺序完全颠倒，当k < n时c为负数，对应循环不会执行，直接导致计算逻辑错误。
• 函数返回逻辑完全失效：末尾使用while k >= n作为返回入口，当输入满足k < n（绝大多数合法组合数输入场景）时，循环条件不成立，函数不会执行任何return语句，默认返回None，这是触发报错/无输出的核心原因。
• 冗余错误逻辑：阶乘计算中判断乘积为0就重置为1/返回1的代码完全无效，Python支持任意精度整数计算，阶乘相乘不会出现溢出归零的情况，这段代码无实际作用。
• 运算精度问题：使用/做除法会返回浮点数，组合数为整数，大数值计算下会出现浮点精度误差。
• 缺失参数边界校验：没有处理k<0、k>n、n为负数等非法输入场景，也没有覆盖k=0、k=n时组合数为1的边界场景。

修正后可运行代码 #

def comb(n, k):
    # 边界参数校验
    if k < 0 or k > n:
        return 0
    if k == 0 or k == n:
        return 1
    # 计算n!
    fact_n = 1
    for i in range(1, n + 1):
        fact_n *= i
    # 计算k!
    fact_k = 1
    for j in range(1, k + 1):
        fact_k *= j
    # 计算(n-k)!
    fact_nk = 1
    c = n - k
    for m in range(1, c + 1):
        fact_nk *= m
    # 整数除法返回准确结果
    return fact_n // (fact_k * fact_nk)

优化提示 #

如果要提升大数值下的计算效率，可以利用组合数性质C(n,k) = C(n, n-k)，始终取k和n-k中更小的值做阶乘计算，减少循环次数；Python 3.10及以上版本也可以直接调用标准库math.comb获取准确结果，不需要自行实现阶乘逻辑。

内容的提问来源于stack exchange，提问作者farzaneh taheri