组合数算计方法 杨辉三角、阶乘逆元、卢卡斯定理

模板题

VJudge: Pink Elephants
输入m, n
求解$C_n^m\%p$

杨辉三角

$0\le m\le n\le 1000$ , $1\le p\le 1e9$ 时使用

int dp[250][250];
const int p = 1000000007;

signed main() {

//#ifdef LOCAL
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
//#endif

    int a, b; cin >> a >> b;
    for(int i = 0; i<220; i++){
        for(int j = 0; j<=i; j++){
            if(j == 0 || i == j) dp[i][j] = 1;
            else dp[i][j] = (dp[i-1][j-1] + dp[i-1][j])%p;
        }
    }
    cout << dp[b][a];
    
    return 0;
}

阶乘逆元

可以用线性复杂度预处理进行优化。
同余定理+逆元: https://blog.csdn.net/LOOKQAQ/article/details/81282342

int fact[250];
const int mod = 1000000007;

int fastPow(int base, int power, int p) {
    int ans = 1;
    while (power > 0) {
        if (power & 1) {
            ans = ans * base % p;
        }
        power >>= 1;
        base = base * base % p;
    }
    return ans;
}

int inv(int a, int p){
    return fastPow(a, p-2, mod);
}


signed main() {

//#ifdef LOCAL
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
//#endif

    int m, n; cin >> m >> n;

    fact[0] = 1;
    for(int i = 1; i<=220; i++){
        fact[i] = i * fact[i-1] % mod;
    }


    cout << fact[n] * inv(fact[m], mod) % mod * inv(fact[n-m], mod) % mod;


    return 0;
}

卢卡斯定理

原理：
$Lucas(n,m,p)modp=Lucas(\frac{n}{p},\frac{m}{p},p)\ast C_{nmodp}^{mmodp}modp$
$Lucas(n,m,p)modp=C_n^{m}modp$

int fact[250];
const int mod = 1000000007;

int fastPow(int a, int b, int p)
{
    int ans = 1;
    a %= p;
    while(b)
    {
        if(b & 1)
        {
            ans = ans * a % p;
            b--;
        }
        b >>= 1;
        a = a * a % p;
    }
    return ans;
}

int C(int n, int m, int p) {
    if (m > n) return 0;
    return (((fact[n] * fastPow(fact[m], p - 2, p)) % p) * fastPow(fact[n - m], p - 2, p)) % p;
}

int lucas(int n, int m, int p) {
    if (!m) return 1;
    return C(n % p, m % p, p) * lucas(n / p, m / p, p) % p;
}


signed main() {

//#ifdef LOCAL
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
//#endif

    int m, n;
    cin >> m >> n;

    fact[0] = 1;
    for (int i = 1; i <= 220; i++) {
        fact[i] = i * fact[i - 1] % mod;
    }

    cout << lucas(n, m, mod);
    
    return 0;
}