codevs 1315 摆花 (DP)

题意:小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共m盆。通过调查顾客的喜好,小明列出了顾客最喜欢的n种花,从1到n标号。为了在门口展出更多种花,规定第i种花不能超过ai盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。

试编程计算,一共有多少种不同的摆花方案。

思路:

dp[i][j] 表示用前i种花摆j天的方案数,状态转移方程:dp[i][j] = (dp[i][j]+dp[i-1][j-k])%mod,  k<=a[i]


代码:

#include
#include
#include
#include
using namespace std;
const int maxn = 1e2+5;
const int INF = 0x3f3f3f3f;
const int mod = 1e6+7;
int dp[maxn][maxn], n, m;
int a[maxn];


int main(void)
{
    while(cin >> n >> m)
    {
        memset(dp, 0, sizeof(dp));
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]);
        for(int i = 0; i <= a[1]; i++)
            dp[1][i] = 1;
        for(int i = 2; 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;
        printf("%d\n", dp[n][m]);
    }
    return 0;
}


你可能感兴趣的:(DP)