注:文章内容源自《挑战程序设计竞赛》(第二版)
原题
多重集组合数
有 n 种物品,第 i 种物品有 ai 个。不同种类的物品可以相互区分但是相同的种类无法区分。从这些物品中取出 m 个的话,有多少种取法?求出方案数模M的余数。
1<=n<=1000
1<=m<=1000
1<=ai<=1000
2<=M<=10000
样例输入
n=3
m=3
a={ 1,2,3 }
M=10000
样例输出
6 (0+0+3,0+1+2,0+2+1,1+0+2,1+1+1,1+2+0)
涉及知识及算法
为了不重复计数,同一种类的物品最好一次性处理好。我们可以这样定义:
dp[i+1][j]:=从前i种物品中取出j个的组合总数
为了从前i种物品中取出j个,可以从前i-1个物品中取出j-k个,再从第i种物品中取出k个添加进来,所以得到如下递推关系:
dp[i+1][j]=∑k=0 min(j,ai) dp[i][j−k]
复杂度为O(nm^2),不过有
dp[i+1][j]=dp[i][j]+dp[i][j-1]+...+dp[i−1][j−ai] ,
所以dp[i+1][j]=dp[i][j]+dp[i+1][j-1]-dp[i][j-ai-1]
这样复杂度就下降到O(nm)了。
代码
int n,m;
int a[MAX_N];
int dp[MAX_N+1][MAX_M+1];
void solve()
{
//一个都不取的方法总是只有一种
for(int i=0;i<=n;i++)
{
dp[i][0]=1;
}
for(int i=0;i<n;i++)
{
for(int j=1;j<=m;j++)
{
if(j-1-a[i]>=0)
{
//在有取余的情况下,要避免减法运算结果出现负数
dp[i+1][j]=(dp[i+1][j-1]+dp[i][j]-dp[i][j-1-a[i]]+M)%M;
}
else
{
dp[i+1][j]=(dp[i+1][j-1]+dp[i][j])%M;
}
}
}
printf("%d\n",dp[n][m]);
}