洛谷——P1077 摆花

一道简单的dp问题。
dp[i][j]表示前i种花j个花盆的方案数。那么就有状态转移方程为sum(dp[i-1][k])，其中k的取值是从0到min(j, a[i])。

#include<iostream>
using namespace std;
int m, n;
int dp[105][105];
int a[105];
int mod = 1000007;
int min(int a, int b) {
	return a < b ? a : b;
}
int main()
{
	cin >> n >> m;
    for (int i = 1; i <= n; i++) cin >> 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 <= min(j, a[i]); k++)
                dp[i][j] = (dp[i][j] + dp[i - 1][j - k]) % mod;
    cout << dp[n][m] << endl;
    return 0;
}

用滚动数组进行优化，dp[j]表示的是摆j盆花的方案数，转化成01背包问题。

#include<iostream>
using namespace std;
int mod = 1000007;
int n, m, a[105], dp[105];
int main() {
    cin>>n>>m;
    for(int i = 1; i <= n; i++) 
        cin>>a[i];
    dp[0] = 1;  //边界条件
    for(int i = 1; i <= n; i++) //枚举每一种花
        for(int j = m; j >= 1; j--)  //枚举每一个摆花的结束点
            for(int k = 1; k <= min(a[i],j); k++)  //枚举摆花的盆数
                dp[j] = (dp[j - k] + dp[j]) % mod;  //求和
    cout<<dp[m];
    return 0;
}