[AcwingXLuogu] 卢卡斯定理/Lucas 定理 模板的使用

前言

luogu传送门 :
Acwing传送门 :

思路

首先给出卢卡斯定理 (就是将求$C(a,b)$分而治之)
$C_a^b\equiv C_{a\%p}^{b\%p}\ast luca{s}_{\frac{a}{p}}^{\frac{b}{p}}(\mathrm{m}\mathrm{o}\mathrm{d}p)$

显然的 : 我们左边$C$是需要求的 而右边$lucas$是通过递归求解
那么左边如何求呢 ?
$C_a^b=\frac{a!}{(a-b)!\ast b!}=\frac{a\ast (a-1)\ast (a-2)\ast \dots \ast (a-b+1)\ast (a-b)\ast \dots \ast 1}{(a-b)\ast (a-b-1)\ast \dots \ast 1\ast b!}=\frac{a\ast (a-1)\ast (a-2)\ast \dots (a-b+1)}{b!}$

我们通过化简可以发现,最后会化简成 每次$\ast a并且/b$ 因此,我们可以通过一层循环

直接求,同时因为需要牵扯到除法取模运算,我们还需要用快速幂求一次逆元

时间复杂度 $O(p+lo{g}_{p}N)$

遗留的问题 :
为什么右边不可以也用组合数求解,而是选择递归的求

CODE

int qmi(int a, int k, int p)
{
    int res = 1;
    while (k)
    {
        if (k & 1) res = (LL)res * a % p;
        a = (LL)a * a % p;
        k >>= 1;
    }
    return res;
}


int C(int a, int b, int p)
{
    if (b > a) return 0;

    int res = 1;
    for (int i = 1, j = a; i <= b; i ++, j -- )
    {
        res = (LL)res * j % p;
        res = (LL)res * qmi(i, p - 2, p) % p;
    }
    return res;
}


int lucas(LL a, LL b, int p)
{
    if (a < p && b < p) 
    return C(a, b, p);
    return (LL)C(a % p, b % p, p) * lucas(a / p, b / p, p) % p;
}


int main()
{
    int n;
    cin >> n;

    while (n -- )
    {
        LL a, b;
        int p;
        cin >> a >> b >> p;
        cout << lucas(a+b, a, p) << endl;
    }

    return 0;
}