887. 求组合数 III

887. 求组合数 III

给定 n 组询问，每组询问给定三个整数 a,b,p，其中 p 是质数，请你输出 Cbamodp 的值。

输入格式

第一行包含整数 n。

接下来 n 行，每行包含一组 a,b,p。

输出格式

共 n 行，每行输出一个询问的解。

数据范围

1≤n≤20,
1≤b≤a≤1e18,
1≤p≤1e5,

输入样例：

3
5 3 7
3 1 5
6 4 13

输出样例：

3
3
2

代码：

/*
Cba=a!/(a−b)!∗b!=a∗(a−1)∗(a−2)∗…∗(a−b+1)∗(a−b)∗…∗1/(a−b)∗(a−b−1)∗…∗1∗b!=a∗(a−1)∗(a−2)∗…(a−b+1)/b!
因此就可以递推的每次乘a然后除以b, 因为从a到a−b+1, 所以就是乘b次
*/
#include <bits/stdc++.h>
using namespace std;

typedef long long LL;

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

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

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

LL lucas(LL a, LL b, int p)
{
    if (a < p && b < p)
        return C(a, b, p);
    return 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, p) << endl;
    }

    return 0;
}
        cin >> a >> b >> p;
        cout << lucas(a, b, p) << endl;
    }

    return 0;
}