将组合数 C n m C_{n}^{m} Cnm进行质因子分解,假设分解结果为 C n m = p 1 c 1 × p 2 c 2 … × p k c k C_{n}^{m}=p_1^{c_1}×p_2^{c_2}…×p_k^{c_k} Cnm=p1c1×p2c2…×pkck,那么 C n m % p C_{n}^{m}\%p Cnm%p就等于 p 1 c 1 × p 2 c 2 … × p k c k % p p_1^{c_1}×p_2^{c_2}…×p_k^{c_k}\%p p1c1×p2c2…×pkck%p,通过快速幂来计算每一组的 p i c i % p p_i^{c_i}\%p pici%p,然后相乘取模就可以得到最终结果,时间复杂度为 O ( k l o g n ) O(klogn) O(klogn),其中 k k k为不超过 n n n的质数个数。适用于 m ≤ n ≤ 1 0 6 m≤n≤10^6 m≤n≤106且 p ≤ 1 0 9 p≤10^9 p≤109。
#include <cstdio>
#include <cmath>
const int maxn = 1000010; //表长
int prime[maxn], pNum = 0; // prime数组存放所有素数,pNum为素数个数
bool p[maxn] = {false}; // 如果i为素数,则p[i]为false,否则p[i]为true
//筛选得到素数表
void Find_Prime() {
for (int i = 2; i < maxn; i++) {
if (p[i] == false) { //如果i是素数
prime[pNum++] = i;
//从2*i开始
for (int j = 2 * i; j < maxn; j += i) {
//筛去所有i的倍数
p[j] = true;
}
}
}
}
//计算n!中有多少个质因子p
int cal(int n, int p) {
int ans = 0;
while (n) {
ans += n / p;
n /= p;
}
return ans;
}
//求a^b%m,递归写法
int BinaryPow(int a, int b, int m) {
if (b == 0) return 1;
// b为奇数,转换为b-1
if (b % 2 == 1) { //可用if (b & 1)替代
return a * BinaryPow(a, b - 1, m) % m;
} else { // b为偶数,转换为b/2
int mul = BinaryPow(a, b / 2, m);
return mul * mul % m;
}
}
//计算C(n,m)%p
int C(int n, int m, int p) {
int ans = 1;
//遍历不超过n的所有质数
for (int i = 0; prime[i] <= n; i++) {
//计算C(n,m)中prime的指数c,cal(n,k)为n!中含质因子k的个数
int c = cal(n, prime[i]) - cal(m, prime[i]) - cal(n - m, prime[i]);
//快速幂计算prime[i]^c%p;
ans = ans * BinaryPow(prime[i], c, p) % p;
}
return ans;
}
int main() {
Find_Prime();
printf("%d", C(60, 34, 200001)); //69886166503903470%200001
return 0;
}