Lucas 是用来求组合数取模的。
Lucas 定理:A、B是非负整数,p是质数。AB写成p进制:A=a[n]a[n-1]…a[0],B=b[n]b[n-1]…b[0]。
可以求逆元来解决,用费马小定理,或者是扩展欧几里得。
给出组合数C(n,m), 表示从n个元素中选出m个元素的方案数。例如C(5,2) = 10, C(4,2) = 6.可是当n,m比较大的时候,C(n,m)很大!于是xiaobo希望你输出 C(n,m) mod p的值!
Input
输入数据第一行是一个正整数T,表示数据组数 (T <= 100) 接下来是T组数据,每组数据有3个正整数 n, m, p (1 <= m <= n <= 10^9, m <= 10^4, m < p < 10^9, p是素数)
Output
对于每组数据,输出一个正整数,表示C(n,m) mod p的结果。
Sample Input
2
5 2 3
5 2 61
Sample Output
1
10
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
long long int n, m, p;
long long int quick_mod(long long int a, long long int b)
{
long long int ans = 1;
a = a%p;
while(b)
{
if(b&1)
ans = ans * a % p;
b>>=1;
a = a * a % p;
}
return ans;
}
long long int C(long long int n, long long int m)
{
if(m>n)
return 0;
long long int ans = 1;
for(int i=1;i<=m;i++)
{
long long int a= (n+i-m)%p;
long long int b = i%p;
ans = ans *(a * quick_mod(b, p-2)%p)%p;
}
return ans;
}
long long int Lucas(long long int n, long long int m)
{
if(m==0)
return 1;
return C(n%p, m%p)*Lucas(n/p, m/p)%p;
}
int main()
{
int t;
cin>>t;
while(t--)
{
scanf("%lld %lld %d", &n, &m, &p);
cout<<Lucas(n, m)<<endl;
}
return 0;
}