题述
有n种物品,第i种物品有ai个。不同种类的物品可以互相区分但相同种类的无法区分。从这些物品中取出m个的话,有多少种取法?求出方案数模M的余数。
限制条件
1<=n<=1000
1<=m<=1000
1<=ai<=1000
2<=M<=10000
样例
输入
3 5
1 2 3
10000
输出
6
题记
dp[i+1][j]:=从前i种物品中取出j个的组合总数
dp[i+1][j]=dp[i+1][j-1]+dp[i][j]-dp[i][j-1-ai]
怎么理解呢,其实跟别的DP题差不多,就是第i种物品取和不取两种情况,但是每种物品可能有多个,所以就涉及到取多少个,这个根据我们做过的题可以想到解决取多少个不需要循环,之前的就已经算过我们可以直接用之前保存的。dp[i+1][j-1]意思就是我们第i个物品至少取1个,也就是当前这一步我要取一个第i个物品(可能已经取过若干个第i个物品了),这就涉及到另外一个问题,如果我们这里的j大于a[i]怎么办,那么我们有可能已经把第i个物品取完了,但是程序不知道啊,它就接着硬着头皮算,所以我们要减去j大于等于a[i]的部分,保证不会取多,因此需要减去一个dp[i][j-1-a[i]],dp[i][j-1-a[i]]也就是我前i-1个物品只取走了j-1-a[i],也就是a[i]会被取走超过a[i]的情况,减一下就行了。
代码如下
#include <iostream>
//多重集组合数
using namespace std;
const int Max=1005;
//输入
int m,n;
int a[Max];
int M;
//dp[i+1][j]从前i种物品中取出j个的组合总数
int dp[Max][Max];
void solve(){
//一个都不取的方法总是只有一种
for(int i=0;i<=n;i++)
dp[i][0]=1;
for(int i=0;i<n;i++){
for(int j=0;j<=m;j++){
if(j-1-a[i]>=0) //如果j大于a[i]存在取第i个物品的数量比该物体的总数还多的情况
dp[i+1][j]=(dp[i+1][j-1]+dp[i][j]-dp[i][j-1-a[i]]+M)%M; //第i个物品至少取一个减去特殊情况再加上第i个物品不取的情况
else
dp[i+1][j]=dp[i+1][j-1]+dp[i][j]%M; //只存在取至少一个或者不取的情况
}
}
printf("%d\n",dp[n][m]);
}
int main()
{
scanf("%d %d",&n,&m);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
scanf("%d",&M);
solve();
return 0;
}