多重组合数-计数类DP

题目：有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;
}