HDU - 3037 Saving Beans

题目链接

题目大意：将不大于m个种子随机放在n个树上，有多少种可能的结果数据量可能有点大，结果模一个素数p

/*

插板法：将N个物品分成M组，每一组至少有一件物品，这个问题就能转换成N个物品摆在桌子上，然后中间有N-1个空，
在这N-1个空里面找M个位置，把板子插进去
*/
/*
但是本题划分的组里面有含有的板子为0的情况所以 每一种的情况为 C(n+i-1, i)
组合数的递推公式: C(n, m) = C(n-1, m-1) + C(n-1, m);
本题的要求是求出来不大于m的所有可能情况：
那么m 可能为0， 1， 2， ... ， m
所以总的答案 = C(n-1, 0) + C(n, 1) + C(n+1, 2) + C(n+2, 3) + ... + C(n+m-1, m)  //迭代累加
             = C(n+m, m)  

*/

AcCode:

//Lacus 求大组合数取余数问题 C(n, m) % p, p一定为素数
//并且p最大为 10^5， m, n可以超级大
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
const int maxn = 100010;
typedef long long LL;
LL factorial[maxn];

LL pow_mod(LL a, LL b, LL p)
{
    LL ans = 1;
    LL base = a;
    while(b){
        if(b & 1) ans = (base*ans) % p;
        base = (base*base) % p;
        b >>= 1;
    }
    return ans;
}

void init(LL p)
{
    factorial[0] = 1;
    for(int i = 1; i <= p; i++)
        factorial[i] = (factorial[i-1]*i) % p;
    return ;
}
LL Lucas(LL n, LL m, LL p)
{
    LL ans = 1;
    while(n && m){
        LL aa = n % p, bb = m % p;
        if(aa < bb) return 0;
        ans = ans * factorial[aa] * pow_mod(factorial[bb]*factorial[aa-bb]%p, p-2, p) % p;
        n /= p, m /= p;
    }
    return ans;
}

int main()
{
    int T;
    LL n, m, p;
    scanf("%d", &T);
    while(T--){
        scanf("%lld %lld %lld", &n, &m, &p);
        init(p);
        LL ans = Lucas(m+n, m, p);
        cout << ans << endl;
    }
    return 0;
}