问题描述
有n种物品,第i种物品有ai个。不同种类的物品可相互区分但相同种类的无法区分,从这些物品中取出m个,有多少种取法?
限制条件
1≤n≤1000
1≤m≤1000
1≤ai≤1000
输入
n=3
m=3
a={1,2,3}
输出
6
建立递推关系
用dp[i][j]表示从前I种物品中取出j个的组合总数,我们可将问题这样分解:若要从前i种物品中取出j个,则需从前i-1种物品中取出j-k个,再从第i种物品中取出k个即可(取法只有一种),由此可得到如下递推关系:
显然直接计算此式的时间复杂度为O(nm2),则可重复利用计算的结果将dp方程压缩。
当j>a[i-1]时,由上式可得:
dp[i][j]=dp[i-1][j]+dp[i-1][j-1]+…dp[i-1][j-ai-1]
dp[i][j-1]=dp[i-1][j-1]+…+dp[i-1][j-1-ai-1]
则结合上述两式得:
dp[i][j]=dp[i][j-1]+dp[i-1][j-1]-dp[i-1][j-1-ai-1]
代码实现
#include <iostream>
#define MAXN 1000
#define MAXM 1000
int n,m;
int a[MAXN];
int dp[MAXN+1][MAXM+1];
int