HDU3037 Saving Beans Lucas 定理+逆元

HDU3037 Saving Beans

题目相当于求n个数的和不超过m的方案数。

如果和恰好等于m，那么就等价于方程x1+x2+...+xn = m的解的个数，利用插板法可以得到方案数为：

(m+1)x(m+2)...(m+n-1)  = C(m+n-1,n-1) = C(m+n-1,m)

现在就需要求不大于m的，相当于对i = 0,1...,m对C(n+i-1,i)求和，根据公式C(n,k) = C(n-1,k)+C(n-1,k-1)得

C(n-1,0)+C(n,1)+...+C(n+m-1,m)

= C(n,0)+C(n,1)+C(n+1,2)+...+C(n+m-1,m)

= C(n+m,m)

现在就是要求C(n+m,m) % p,其中p是素数。

然后利用Lucas定理的模板就可以轻松的求得C(n+m,m) % p的值

#include<bits/stdc++.h>
using namespace std;
const int MAX=1e5+1;
long long n,m,MOD;
long long fact[MAX];
inline long long inv_(long long a)
{
    if(a==1) return 1;
    return inv_(MOD%a)*(MOD-MOD/a)%MOD;
}
inline void init()
{
    fact[0]=1;
    for(int i=1;i<=MOD;i++) fact[i]=(fact[i-1]*i)%MOD;
}
inline long long C(long long a,long long b)
{
    if(b>a) return 0;
    return fact[a]*inv_(fact[a-b]*fact[b]%MOD)%MOD;
}
inline long long lucas(long long a,long long b)
{
    if(a<MOD&&b<MOD) return C(a,b);
    return lucas(a/MOD,b/MOD)*C(a%MOD,b%MOD)%MOD;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%I64d%I64d%I64d",&n,&m,&MOD);
        init();
        printf("%I64d\n",lucas(n+m,m));
    }
    return 0;
}