zcmu 1549 : 조합 수 (루카스 정리)

제목 링크 : https://acm.zcmu.edu.cn/JudgeOnline/problem.php?id=1549

이야기

n, m, p를주고 C (n, m) % p의 조합 수를 찾으십시오.

범위 : (1 <= m <= n <= 10 ^ 9, m <= 10 ^ 4, 0 <p <100, p는 소수)

아이디어

n과 m의 범위는 매우 큽니다 .C를 직접 계산하면 t가 얻어 지지만 여기서는 계수 p가 작기 때문에 p로 시작할 수 있습니다.

C (n, m) = n! * (n-m)! % p * m! % p

루카스 정리 적용 : 음이 아닌 정수 m, n 및 소수 p

, m과 n의 p 항 확장은 어디에 있습니까?

n <m 일 때 규정

ac 코드

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 105;
ll fac[maxn], inv[maxn], p;
// fac[i]表示i的阶乘对p取模
// inv[i]表示i的阶乘的逆元对p取模
void init(ll n){
    fac[0] = 1;
    for(int i = 1; i <= n; i ++) fac[i] = fac[i - 1] * i % p;
    inv[n] = _pow(fac[n], p - 2);
    for(int i = n - 1; i >= 0; i --) inv[i] = inv[i + 1] * (i + 1) % p;
}
ll C(ll n, ll m){
    if(n < m) return 0;
    return fac[n] * inv[m] % p * inv[n - m] % p;
}
ll lucas(ll n, ll m){//卢卡斯定理
    if(m == 0) return 1 % p;
    return lucas(n / p, m / p) * C(n % p, m % p) % p;
}
ll _pow(ll a, ll b){
    ll ans = 1;
    while(b){
        if(b & 1) ans = ans * a % p;
        a = a * a % p;
        b >>= 1;
    }
    return ans;
}
int main(){
    int t; scanf("%d", &t);
    while(t --){
        ll n, m;
        scanf("%lld%lld%lld",&n,&m,&p);
        init(p - 1);
        printf("%lld\n", lucas(n, m));
    }
    return 0;
}