题意:小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共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<iostream> #include<cstdio> #include<cstring> #include<cmath> 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; }