HDU 5201 The Monkey King

题意：给定n个桃子和m个猴子（其中一个孙悟空）现求孙悟空桃子数量严格大于其他猴子桃子数量的方案数

首先枚举孙悟空的桃子数i（从n/.m到n） 不考虑“大于”这个条件的方案数为C（n-i+m-1-1,m-1-1） (隔板法) 减去非法状况即可

设 f(j) 表示保证有j个以上的猴子桃子数量比孙悟空多 这里的做法可以理解为 先取出j*i个桃子预先给j个猴子(一会还会分到桃) (j个猴子由(c(m-1，j))得到)

 然后插板法把剩下的桃子在分给m-1个猴子 f(j) 得到了  就可以容斥求出 所以非法的状况了即sum( (-1)^j *f(j) )

注意特判下m=1的时候(＞﹏＜)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#define scnaf scanf
#define cahr char
#define bug puts("bugbugbug");
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
const int mod=1e9+7;
const int maxn=2e5+100;
const int inf=1e9;
ll pow_mod(ll a,ll b,int mod)
{
    ll ans=1;
    a%=mod;
    while(b)
    {
        if(b&1) ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
ll fac[maxn+10],inv[maxn+10];
void init()
{
    fac[0]=1;
    for(int i=1; i<maxn; i++)
        fac[i]=fac[i-1]*i%mod;
    inv[maxn-1]=pow_mod(fac[maxn-1],mod-2,mod);
    for(int i=maxn-2; i>=0; i--)
        inv[i]=inv[i+1]*(i+1)%mod;
}
ll c(int a,int b)
{
    if(a<b)return 0;
//    if(a<0||b<0)return 0;
    return fac[a]*inv[b]%mod*inv[a-b]%mod;
}
int main()
{
    init();
    int T_T,n,m;
    scanf("%d",&T_T);
    while(T_T--)
    {
        scanf("%d%d",&n,&m);
        if(m==1) {
            puts("1");
            continue;
        }
        ll ans=0;
        for(int i=n/m; i<=n; i++) //枚举齐天大圣的桃子数量
        {
            ll nn=n-i,mm=m-1;
            ans+=c(nn+mm-1,mm-1);//剩下的n-i个桃子分给剩下的m-1个猴子
            ans%=mod;
            for(int j=1; j<=mm&&i*j<=nn; j++) //保证有j个猴子的桃子数量大于大圣的
            {
                int fh=1;
                if(j%2) fh=-1;
                ans+=mod+c(mm,j)*c(nn-j*i+mm-1,mm-1)%mod*fh;
                ans%=mod;
            }
        }
        printf("%lld\n",ans%mod);
    }
    return 0;
}