卢卡斯定理/Lucas 定理

题目描述

给定整数 n, m, p 的值，求出 Cn+m n​（mod）p 的值。

输入数据保证 p 为质数。

注: C 表示组合数。

输入格式

本题有多组数据。

第一行一个整数 T，表示数据组数。

对于每组数据:

一行，三个整数 n, m, p。

输出格式

对于每组数据，输出一行，一个整数，表示所求的值。

输入输出样例

输入 #1复制

2
1 2 5
2 1 5

输出 #1复制

3
3

说明/提示

对于 100\%100% 的数据，1≤n,m,p≤10^5，1≤T≤10。

#include <iostream>
#include <algorithm>

using namespace std;

typedef long long LL;


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, b, p) << endl;
    }

    return 0;
}