P3807 【模板】卢卡斯定理/Lucas 定理

【模板】卢卡斯定理/Lucas 定理

题目背景

这是一道模板题。

题目描述

给定整数 $n,m,p$ 的值，求出 $C_{n+m}^n\mathrm{mod}p$ 的值。

输入数据保证 $p$ 为质数。

注: $C$ 表示组合数。

输入格式

本题有多组数据。

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

对于每组数据:

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

输出格式

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

样例 #1

样例输入 #1

2
1 2 5
2 1 5

样例输出 #1

3
3

提示

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

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e5 + 10;
int a[N];
int qpow(int a, int k, int p) {
    a %= p;
    int res = 1ll;
    while (k) {
        if (k & 1) res = res * a % p;
        a = a * a % p;
        k >>= 1;
    }
    return res;
}
int p;
int C(int n, int m) {
    if (m > n) return 0;
    return (a[n] * qpow(a[m], p - 2, p)) % p * qpow(a[n - m], p - 2, p) % p;
}
int lucas(int n, int m) {
    if (!m) return 1;
    return C(n % p, m % p) * lucas(n / p, m / p) % p;
}
int n, m, t;
signed main() {
    cin >> t;
    while (t -- ) {
        cin >> n >> m >> p; int o = max(n, m); a[0] = 1;
        for (int i = 1; i <= p; i ++ )
            a[i] = a[i - 1] * i % p;
        cout << lucas(n + m, m) % p << '\n';
    }
}