NKOJ4052 糖果盒

【题解】【排列组合】
问题描述

何老板有很多颗相同的糖果，同时还有n个不同的糖果盒。何老板想把其中一些糖果放入这些盒子里，要求所有盒子里总的糖果数不超过m，问总共有多少种不同的方案？
每个盒子的容量无限大，可以选择装糖，也可以不装。
答案可能很大，mod p后在输出，p是一个素数。

输入格式

第一行，三个空格间隔的整数 n,m,p （1 <= n, m <= 100000000, 1 < p < 100000）

输出格式

一行，一个整数，表示计算结果

把相同的糖果分成若干堆，显然是隔板法。但是题目中要求可以有空盒子，直接使用隔板法就行不通了。

技巧：设置几个虚拟的糖果，数量和盒子的数量一致。

盒子里糖果数量为m的情况：

这个时候，糖果总数就为n+m，有n+m-1个空和n-1块板，所以方案数为C(n+m-1,n-1),化简得到C(n+m-1,m)。

盒子里糖果数量不超过m的情况：

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

结论：方案总数为C(n+m,m)。

又因为mod一个质数，所以最后可用卢卡斯定理求得答案。

总结

1.把相同的元素分成若干个集合，用隔板法。

2.把相同的元素分成若干个集合，允许有空集合，设置虚拟元素。

3.组合式子比较复杂时，要化简

4.不要漏了0颗糖果的情况（一个元素都没有）

代码

#include<iostream>
#include<cstdio>
#define LL long long
using namespace std;
LL mont(LL a,LL b,LL c){
    LL ans=1;a%=c;
    while(b){
        if(b&1)ans=(ans*a)%c;
        b>>=1,a=(a*a)%c;
    }
    return ans%c;
}
LL C(LL a,LL b,LL p){
    if(a==b)return 1;
    if(b>a-b)b=a-b;
    LL A=1,B=1;
    for(LL i=0;i<b;i++){
        A=(A*(a-i))%p;
        B=(B*(b-i))%p;
    }
    return (A*(mont(B,p-2,p)))%p;
}
int main(){
    LL n,m,p;
    scanf("%lld%lld%lld",&n,&m,&p);
    printf("%lld",C(n+m,m,p));
    return 0;
}