FZU 2020 组合（Lucas定理的应用）

Problem 2020 组合

Accept: 530    Submit: 1323
Time Limit: 1000 mSec    Memory Limit : 32768 KB

Problem Description

给出组合数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

25 2 35 2 61

Sample Output

110 

AC代码：

#include <stdio.h>
#define ll long long

int qp(ll a,ll n, ll p){//快速幂
	ll ans=1;
	while(n)
	{
		if(n&1)
			ans=(ans*a)%p;
		n>>=1;
		a=(a*a)%p;
	}
	return ans;
}

int C(ll n,ll m,ll p){
	ll a,b;
	a=b=1;
	if(m>n)
		return 0;
	while(m){
		a=(a*n)%p;
		b=(b*m)%p;
		m--;
		n--;
	}
	return (a*qp(b,p-2,p))%p;//利用逆元求C(n,m)%p;
}

int Lucas(ll n,ll m,ll p){
	if(m==0)
		return 1;
	return (C(n%p,m%p,p)*Lucas(n/p,m/p,p))%p;
}

int main(){
	int n,m,p,t;
	scanf("%d",&t);
	while(t--){
		scanf("%d%d%d",&n,&m,&p);
		printf("%d\n",Lucas(n,m,p));
	}
	return 0;
}