(n, k)的二项式系数由以下公式计算:
(n, k) = n! / k! / (n - k)!
要使此功能适用于大量 n 和 k 模 m 请注意:
模数 m 的阶乘可以逐步计算,在
每一步都会得到结果% m。但是,如果 n 达到 10^18,这将太慢。因此,faster methods 的复杂性受模数限制,您可以使用其中的一些。
除法(a / b) mod m 等于(a * b^-1) mod m,其中b^-1 是b 模m 的倒数(即(b * b^-1 = 1) mod m)。
这意味着:
(n, k) mod m = (n! * (k!)^-1 * ((n - k)!)^-1) mod m
使用Extended Euclidean algorithm 可以有效地找到数字的倒数。假设您已经解决了阶乘计算,那么算法的其余部分很简单,只需注意乘法时的整数溢出。这是适用于n=10^9 的参考代码。为了处理更大的数字,应该用更有效的算法替换阶乘计算,并且应该稍微调整代码以避免整数溢出,但主要思想将保持不变:
#define MOD 1000000007
// Extended Euclidean algorithm
int xGCD(int a, int b, int &x, int &y) {
if (b == 0) {
x = 1;
y = 0;
return a;
}
int x1, y1, gcd = xGCD(b, a % b, x1, y1);
x = y1;
y = x1 - (long long)(a / b) * y1;
return gcd;
}
// factorial of n modulo MOD
int modfact(int n) {
int result = 1;
while (n > 1) {
result = (long long)result * n % MOD;
n -= 1;
}
return result;
}
// multiply a and b modulo MOD
int modmult(int a, int b) {
return (long long)a * b % MOD;
}
// inverse of a modulo MOD
int inverse(int a) {
int x, y;
xGCD(a, MOD, x, y);
return x;
}
// binomial coefficient nCk modulo MOD
int bc(int n, int k)
{
return modmult(modmult(modfact(n), inverse(modfact(k))), inverse(modfact(n - k)));
}