Python求组合数C(n, k)的值及进阶求（x + 1）^n的展开式系数

最新推荐文章于 2024-04-09 16:25:33 发布

拙诚杰

最新推荐文章于 2024-04-09 16:25:33 发布

阅读量418

点赞数 1

文章标签： python 开发语言

本文链接：https://blog.csdn.net/m0_74549061/article/details/131485075

版权

1.求组合数C(n, k)的值（函数）

#组合数C(n, k)的计算
def combination(n, k):
    if k < 0 or n <= 0 or k > n: #不满足定义的情况
        print("error!")
    elif k == 0: #特殊情况
        print(1)
    else:
        x = 1
        y = 1
        for i in range(n - k + 1, n + 1): #分子
            x = x * i
        for j in range(1, k + 1): #分母
            y = y * j
        print(int(x / y))
combination(10, 4)#以C(10. 4)为例，输出结果为210

2.求展开式系数

展开 (x + 1)^n 的式子可以使用二项式定理来计算。根据二项式定理，展开式的每一项可以表示为：

C(n, k) * x^(n-k) * 1^k

其中，C(n, k) 表示组合数，计算方式为 n! / (k! * (n-k)!)， x^(n-k) 表示 x 的 n-k 次幂， 1^k 表示 1 的 k 次幂。

以下是展开式的完整表达式：

(x + 1)^n = C(n, 0) * x^(n-0) * 1^0 + C(n, 1) * x^(n-1) * 1^1 + C(n, 2) * x^(n-2) * 1^2 + ... + C(n, n-1) * x^(n-(n-1)) * 1^(n-1) + C(n, n) * x^(n-n) * 1^n

简化后，展开式可表示为：

(x + 1)^n = C(n, 0) * x^n + C(n, 1) * x^(n-1) + C(n, 2) * x^(n-2) + ... + C(n, n-1) * x + C(n, n)

其中，每一项的系数 C(n, k) 可以使用组合数的公式计算。

于是我们可以使用1中求组合数C(n, k)的函数继续完成展开式的计算

代码如下：

#1中的函数代码
def combination(n, k):
    if k < 0 or n <= 0 or k > n: #不满足定义的情况
        print("error!")
    elif k == 0: #特殊情况
        print(1)
    else:
        x = 1
        y = 1
        for i in range(n - k + 1, n + 1): #分子
            x = x * i
        for j in range(1, k + 1): #分母
            y = y * j
        print(int(x / y))

#主代码
n = input()
n = int(n)
for o in range(0, n + 1):
    combination(n, o) #调用函数

输出结果为：1 4 6 4 1