Python 空间和时间高效的二项式系数(Space and time efficient Binomial Coefficient)

这里函数采用两个参数n和k,并返回二项式系数 C(n, k) 的值。 

例子: 

输入: n = 4 和 k = 2

输出: 6

解释: 4 C 2 等于 4!/(2!*2!) = 6

输入: n = 5 和 k = 2

输出: 10

解释: 5 C 2 等于 5!/(3!*2!) = 10

        在本文中,我们讨论了 O(n*k) 时间和 O(k) 额外空间算法。C(n, k) 的值可以在 O(k) 时间和 O(1) 额外空间内计算出来。

方法:

1、如果 r 大于 nr,则将 r 更改为 nr,并创建一个变量来存储答案。

2、从 0 到 r-1 运行循环

3、在每次迭代中更新 ans 为 (ans*(ni))/(i+1),其中 i 是循环计数器。

4、所以答案将等于 ((n/1)*((n-1)/2)*…*((n-r+1)/r),等于 nCr。

C(n, k) 
= n! / (nk)! * k! 
= [n * (n-1) *....* 1] / [ ( (nk) * (nk-1) * .... * 1) * 
                            ( k * (k-1) * .... * 1 ) ]
简化后,我们得到
C(n, k) 
= [n * (n-1) * .... * (n-k+1)] / [k * (k-1) * .... * 1]

另外,C(n, k) = C(n, nk)   
// 如果 r > n-r,则 r 可以更改为 n-r

以下实现中利用上述公式计算C(n,k):

# Python program to calculate C(n, k) 
  
# Returns value of Binomial Coefficient 
# C(n, k) 

def binomialCoefficient(n, k): 
    
    # since C(n, k) = C(n, n - k) 
    
    if(k > n - k): 
        k = n - k 
    
    # initialize result 
    res = 1
    
    # Calculate value of  
    # [n * (n-1) *---* (n-k + 1)] / [k * (k-1) *----* 1] 
    for i in range(k): 
        res = res * (n - i) 
        res = res // (i + 1) 
    return res 
  
# Driver program to test above function  

n = 8

k = 2

res = binomialCoefficient(n, k) 

print("Value of C(% d, % d) is % d" %(n, k, res)) 
  
# This code is contributed by Aditi Sharma 

输出:
C(8, 2) 的值为 28

复杂度分析: 

时间复杂度: O(r)循环必须从 0 运行到 r。因此,时间复杂度为 O(r)。

辅助空间:O(1),因为不需要额外的空间。

  • 11
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值