%mod兼容生成二项式系数的方法

时间:2012-02-20 23:58:43

标签: c++ algorithm binomial-coefficients

我想优化我的程序的一部分,我计算二项式系数之和到K.即

C(N,0) + C(N,1) + ... + C(N,K)

由于值超出数据类型(long long)可以支持,我将计算值mod M并且正在寻找执行此操作的过程。

目前,我已经完成了Pascal的三角形,但似乎需要一点负担。所以,我想知道是否有其他有效的方法来做到这一点。我已经考虑过卢卡斯的定理,虽然我已经足够大,所以C(N,k)失控了!

任何指针,如何以不同的方式做到这一点,也许可以用一些其他整齐的表达来计算整数和。如果不是,我将把它与Pascal的Triangle方法本身一起留下。

谢谢,

以下是我到目前为止O(N^2)

#define MAX 1000000007
long long NChooseK_Sum(int N, int K){
    vector<long long> prevV, V;
    prevV.push_back(1);     prevV.push_back(1);
    for(int i=2;i<=N;++i){
            V.clear();
            V.push_back(1);
            for(int j=0;j<(i-1);++j){
                    long long val = prevV[j] + prevV[j+1];
                    if(val >= MAX)
                            val %= MAX;
                    V.push_back(val);
            }
            V.push_back(1);
            prevV = V;
    }
    long long res=0;
    for(int i=0;i<=K;++i){
            res+=V[i];
            if(res >= MAX)
                    res %= MAX;
    }
    return res;
}

1 个答案:

答案 0 :(得分:5)

执行线性数量的算术bignum操作的算法是

def binom(n):
    nck = 1
    for k in range(n + 1):  # 0..n
        yield nck
        nck = (nck * (n - k)) / (k + 1)

这使用除法,但以p为模,你可以通过将解i乘以等式i * (k + 1) = 1 mod p来完成同样的事情。可以通过extended Euclidean algorithm在对数个算术运算中找到值i