题目:有n种物品,第i种物品有p[i]个,不同种类的物品可以相互区分,同种类的物品不能相互区分。
从这些物品种取出m个,有多少种取法,答案对mod取模。
思想:
dp[i][j]表示前i种物品,一共拿了j个物品的方案数。
为了得到dp[i][j],那么可以从前i-1种物品取j-k个,再从第i种物品取k个即可
下面的x[i]替换成p[i]
分2种情况讨论:
情况1:j<=x[i]
右边展开得到的是dp[i-1][0] dp[i-1][1] dp[i-1][2]....dp[i-1][j] 、 把最后一项拿出来,剩下的j-1项求和,但是我们写成这样:
,与原来等价,且我们能轻易发现,这就是dp[i][j-1];
也就是 dp[i][j]=dp[i][j-1]+dp[i-1][j];
情况 2: j>x[i]
右边展开得到的是dp[i-1][j-x[i]] dp[i-1][j-x[i]+1] dp[i-1][j-x[i]+2]....dp[i-1][j] 、仿照上面的形式,同样得到一个求和式(上界取x[i]),
显然能发现这个式子拆开后,多出一项dp[i][j-1-x[i]],所以最后要减掉,即:dp[i][j]=dp[i][j-1]+dp[i-1][j] - dp[i][j-1-x[i]];
综上,递推式为:
if(j >p[i] )
dp[i+1][j] = (dp[i][j] + dp[i+1][j-1] - dp[i][j-1-p[i]] +mod)%mod;
else
dp[i+1][j] = dp[i][j] + dp[i+1][j-1];
代码模板:
void solve()
{
int n,m;
cin>>n>>m;
for(int i=0; i<n; ++i)
{
cin>>p[i];
}
cin>>mod;
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>p[i])
{
dp[i+1][j]=(dp[i][j]+dp[i+1][j-1]-dp[i][j-1-p[i]]+mod)%mod;
}
else
{
dp[i+1][j]=dp[i][j]+dp[i+1][j-1];
}
}
}
cout<<dp[n][m]<<endl;
}