多重集组合数

题目: 有n种物品, 第i种物品有a个. 不同种类的物品可以互相区分, 但相同种类的无法区分. 从这些物品中取出m个, 有多少种取法?

求出数模M的余数.

例如: 有n=3种物品, 每种a={1,2,3}个, 取出m=3个, 取法result=6(0+0+3, 0+1+2, 0+2+1, 1+0+2, 1+1+1, 1+2+0).

dp[i][j] 表示前i种物品,一共拿了j个物品的方法数

为了得到dp[i][j]，那么可以从前i-1种物品取j-k个,再从第i种物品取k个即可

即 有：

然而这个公式是O（n （m^2））的。

不过我们可以发现，把右边求和式展开

分两种情况：

情况1：j<=x[i] （即j-1<x[i]）

右边展开得到的是dp[i][0] dp[i][1] dp[i][2]…dp[i][j] 、 把最后一项拿出来，剩下的j-1项求和，但是我们写成这样：

，与原来等价，且我们能轻易发现，这就是dp[i][j-1];

也就是 dp[i][j]=dp[i][j-1]+dp[i-1][j];

情况 2： j>x[i]

右边展开得到的是dp[i][j-x[i]] dp[i][j-x[i]+1] dp[i][j-x[i]+2]…dp[i][j] 、仿照上面的形式，同样得到一个求和式（上界取x[i]），

显然能发现这个式子拆开后，多出一项dp[i-1][j-1-x[i]]，所以最后要减掉，

即：
dp[i][j]=dp[i][j-1]+dp[i-1][j] - dp[i-1][j-1-x[i]];

综上，递推式为：

注意此时 i 变为 i + 1

if(j >a[i] )
dp[i+1][j] = (dp[i][j] + dp[i+1][j-1] - dp[i][j-1-a[i]] +M)%M;
else{
dp[i+1][j] = dp[i][j] + dp[i+1][j-1];

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <queue>
#include <map>
#include <set>
#include <vector>
using namespace std;
int n,m;
int a[1005];
 
int dp[1005][1005];
 
int main()
{
	while(cin >> n >> m){
		for(int i = 0;i < n;i++){
			cin >> a[i];
		}
		int M;
		
		cin >> M;
		for(  i = 0;i <= n;i++){
			dp[i][0] = 1;
		}
		for(  i = 0;i < n;i++){
			for(int j = 1;j <= m;j++){
				if(j >a[i] )
					dp[i+1][j] = (dp[i][j] + dp[i+1][j-1] - dp[i][j-1-a[i]] +M)%M;
//此处+M是防止减法操作得到一个负数， 加一个M不影响结果并保证了答案不为负数。
				else{
					dp[i+1][j] = dp[i][j] + dp[i+1][j-1]; 
				}
			}
		} 
	printf("%d\n",dp[n][m]);
	} 	
	return 0;
	
}

————————————————
版权声明：本文为CSDN博主「yuhong_liu」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/viphong/article/details/48110525