摆花
描述
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共 m 盆。通过调查顾客的喜好,小明列出了顾客最喜欢的 n 种花,从 1 到 n 标号。为了在门口展示出更多种花,规定第i种花不能超过a i 盆,同一种花摆放在一起,且不同种类的花需按标号从小到大的顺序依次摆列。 试编程计算,一共有多少种不同的摆花方案。
输入
第 1 行包含 2 个正整数 n 和 m,中间用一个空格隔开。第 2 行有 n 个整数依次表示 a i ,每两个整数之间用一个空格隔开。
输出
输出一行一个整数,表示有多少种摆放方案。注意:因为方案数可能很多,请输出方案数对 1000007 取模的结果。
输入样例 1
2 4 3 2
输出样例 1
2
#include <iostream>
#include <vector>
using namespace std;
const int MOD = 1000007;
int main() {
int n, m;
cin >> n >> m;
vector<int> a(n);
for (int i = 0; i < n; ++i) {
cin >> a[i];
}
vector<vector<int>> dp(n+1, vector<int>(m+1, 0));
dp[0][0] = 1;
for (int i = 1; i <= n; ++i) {
for (int j = 0; j <= m; ++j) {
for (int k = 0; k <= min(a[i-1], j); ++k) {
dp[i][j] = (dp[i][j] + dp[i-1][j-k]) % MOD;
}
}
}
cout << dp[n][m] << endl;
return 0;
}