洛谷 P1077摆花

洛谷 P1077摆花

题目描述
小明的花店新开张，为了吸引顾客，他想在花店的门口摆上一排花，共m盆。通过调查顾客的喜好，小明列出了顾客最喜欢的n种花，从1到n标号。为了在门口展出更多种花，规定第ii种花不能超过$a_i$ 盆，摆花时同一种花放在一起，且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算，一共有多少种不同的摆花方案。

思路：这道题其实思路不难，很类似于背包，状态也很好定义：$dp[i][j]$表示摆到i号花已经摆了j盆的方案数，那么转移就很自然了：
$$dp[i][j]+=dp[i-1][j-k],(0\le k\le a[i])$$

那么我为啥还要写这个题解呢？(因为WA)
原因就是边界的设置！！！
请看10分代码

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int mod=1000007;
int n,m,a[1000],dp[1000][1000];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i) 
    {
        scanf("%d",&a[i]);
        dp[i][0]=1;
    }
    for(int i=2;i<=n;++i)
    {
        for(int j=0;j<=m;++j)
        {
            for(int k=0;k<=a[i];++k)
            {
                if(j-k>=0)
                {
                    dp[i][j]=(dp[i][j]+dp[i-1][j-k]%mod)%mod;
                }
            }
        }
    }
    printf("%d",dp[n][m]);
    return 0;
    
}

为啥不对呢？因为边界设置的不对，边界应该是$dp[0][0]=1$
改后AC代码

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int mod=1000007;
int n,m,a[1000],dp[1000][1000];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i) scanf("%d",&a[i]);
    dp[0][0]=1;
    for(int i=1;i<=n;++i)
    {
        for(int j=0;j<=m;++j)
        {
            for(int k=0;k<=a[i];++k)
            {
                if(j-k>=0)
                {
                    dp[i][j]=(dp[i][j]+dp[i-1][j-k]%mod)%mod;
                }
            }
        }
    }
    printf("%d",dp[n][m]);
    return 0;
    
}

其实我到现在也没有完全理解为啥是$dp[0][0]$，但不得不说，大部分方案数dp问题的边界都是$dp[0][0]=1$，所以就当是积累一下经验吧