【hdu3037】【Lucas定理】Saving Beans

经过推导答案为C(n + m,m)

但是n和m的范围相当大，所以我们可以使用Lucas定理，它可以用来解决大组合数取模的问题。

定理的内容:

A、B是非负整数，p是质数。AB写成p进制：A=a[n]a[n-1]...a[0]，B=b[n]b[n-1]...b[0]。
则组合数C(A,B)与C(a[n],b[n])*C(a[n-1],b[n-1])*...*C(a[0],b[0])  modp同余

即：Lucas(n,m,p)=c(n%p,m%p)*Lucas(n/p,m/p,p) 

这样上面的问题就被很好的解决了。

代码:

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 100000 + 10;
int T;
long long n,m,p;
long long fac[maxn];

void init()
{
	freopen("hdu3037.in","r",stdin);
	freopen("hdu3037.out","w",stdout);
}

void prepare(long long p)
{
	fac[0] = 1;
	for(int i = 1;i <= p;i++)fac[i] = fac[i-1] * i % p;
}

long long quick_pow(long long a,long long b,long long p)
{
	long long ret = 1,tmp = a;
	while(b)
	{
		if(b & 1)ret = ((ret % p) * (tmp % p)) % p;
		tmp = ((tmp % p) * (tmp % p)) % p;
		b >>= 1;
	}
	return ret;
}

long long Lucas(long long a,long long b,long long p)
{
	long long ret = 1;
	while(a && b)
	{
		long long adig = a % p,bdig = b % p;
		if(adig < bdig)return 0;
		ret = ret * fac[adig] * quick_pow((fac[bdig] * fac[adig-bdig]) % p,p - 2,p) % p;
		a /= p;b /= p;
	}
	return ret;
}

void solve()
{
	prepare(p);
	printf("%lld\n",Lucas(n + m,m,p));
}

void readdata()
{
	scanf("%d",&T);
	while(T--)
	{
		scanf("%lld%lld%lld",&n,&m,&p);
		solve();
	}
}

int main()
{
	init();
	readdata();
	return 0;
}