小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共 mm 盆。
通过调查顾客的喜好,小明列出了顾客最喜欢的 n 种花,从 1 到 n 标号。
为了在门口展出更多种花,规定第 i 种花不能超过 ai 盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算,一共有多少种不同的摆花方案。
输入格式
第一行包含两个正整数 n 和 m,中间用一个空格隔开。
第二行有 nn 个整数,每两个整数之间用一个空格隔开,依次表示 a1,a2,…,an。
输出格式
输出只有一行,一个整数,表示有多少种方案。
注意:因为方案数可能很多,请输出方案数对 1000007 取模的结果。
数据范围
0<n,m≤100,
0≤ai≤100
输入样例:
2 4
3 2
输出样例:
2
(DP,多重背包问题,背包问题求方案数) O(n2a)
问题转化:
将花盆数量看作背包容量;
将花看作物品,体积是1,第 i 种物品最多选 ai 个;
问题:将背包装满的方案数是多少?
这是典型的多重背包求方案数问题:
状态表示:f[i, j] 表示前i个物品,总体积是j的方案数;
状态计算:f[i, j] = f[i - 1, j] + f[i - 1, j - 1] + ...
+ f[i - 1, j - a[i]]。
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 110, mod = 1000007;
int n, m;
int f[N];
int main()
{
cin >> n >> m;
f[0] = 1;
for (int i = 0; i < n; i ++ )
{
int a;
cin >> a;
for(int j=m;j>=0;j--){
for(int k=1;k<=min(a,j);k++){//从1开始
f[j] = (f[j] + f[j - k]) % mod;//最开始的f[j]表示f[i-1][j-0];
}
}
}
cout<<f[m]<<endl;
return 0;
}
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 110, mod = 1000007;
int n, m;
int f[N][N];
int main()
{
cin >> n >> m;
f[0][0] = 1;
for (int i = 1; i <= n; i ++ )
{
int a;
cin >> a;
for(int j=0; j<=m; j++){
for(int k=0;k<=min(a,j);k++){
f[i][j] = (f[i][j] + f[i-1][j - k]) % mod;
}
}
}
cout<<f[n][m]<<endl;
return 0;
}