FZUOJ2020 组合 (lucas定理)

 Problem 2020 组合

Accept: 705    Submit: 1706
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

 Source

FOJ有奖月赛-2011年04月（校赛热身赛）

解析：裸的lucas定理。

代码：

#include<cstdio>
#include<algorithm>
using namespace std;

typedef long long LL;

LL reverse(LL a,LL m)
{
  if(a==0)return 0;
  LL y=0,x=1,r=a%m,q,t,mm=m;
  if(r<0)r+=m;
  while((m%r)!=0)
    {
      a=m,m=r,q=a/m,r=a%m;
      t=x,x=y-x*q,y=t;
	}
  if(r!=1)return 0;
  if(x<0)x+=mm;
  return x;
}

LL C(LL n,LL m,LL p)
{
  LL i,j,k=min(m,n-m),ans=1;
  for(i=n-k+1;i<=n;i++)ans=ans*i%p;
  for(i=2;i<=k;i++)ans=ans*reverse(i%p,p)%p;
  return ans;
}

LL 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()
{
  LL i,j,k,t,n,m,p;
  scanf("%I64d",&t);
  while(t--)
    {
      scanf("%I64d%I64d%I64d",&n,&m,&p);
      printf("%I64d\n",lucas(n,m,p));
	}
  return 0;
}