求组合数Ⅲ

题目链接
题意：给定n组询问，每组询问给定三个整数a,b,p，其中p是质数，请你输出Cba mod p的值。
输入格式
第一行包含整数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
思路：直接利用卢卡斯定理求解

代码实现： O(p * logN * log p)

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 100010, mod = 1e9 + 7;
int p;

int qmi(int a, int k)
{
    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 res = 1;
    for(int i = 1, j = a; i <= b; i ++, j --){
        res = (ll)res * j % p;
        //除i等于乘以i的逆元，费马定理i模p的逆元就是i ^(p-2)
        res = (ll)res * qmi(i, p - 2) % p;
    }
    return res;
}

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

int main()
{
    int n; cin >> n;
    while(n --){
        ll a, b;
        cin >> a >> b >> p;
        cout << lucas(a, b) << endl;
    }
    return 0;
}