【题解】【排列组合】
问题描述
何老板有很多颗相同的糖果,同时还有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;
}