题目描述
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共 mm 盆。通过调查顾客的喜好,小明列出了顾客最喜欢的 nn 种花,从 11 到 nn 标号。为了在门口展出更多种花,规定第 ii 种花不能超过 a_iai 盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。
试编程计算,一共有多少种不同的摆花方案。
输入格式
第一行包含两个正整数 nn 和 mm,中间用一个空格隔开。
第二行有 nn 个整数,每两个整数之间用一个空格隔开,依次表示 a_1,a_2, \cdots ,a_na1,a2,⋯,an。
输出格式
一个整数,表示有多少种方案。注意:因为方案数可能很多,请输出方案数对 10^6+7106+7 取模的结果。
输入输出样例
输入
2 4 3 2
输出
2
说明/提示
【数据范围】
对于 20\%20% 数据,有 0<n \le 8,0<m \le 8,0 \le a_i \le 80<n≤8,0<m≤8,0≤ai≤8。
对于 50\%50% 数据,有 0<n \le 20,0<m \le 20,0 \le a_i \le 200<n≤20,0<m≤20,0≤ai≤20。
对于 100\%100% 数据,有 0<n \le 100,0<m \le 100,0 \le a_i \le 1000<n≤100,0<m≤100,0≤ai≤100。
二维dp方法:
1.确定dp方程意义dp[n][m],第n种花总共放m盆。
2.确定动态转移方程,dp[i][j]=dp[i-1][j-k];
3.初始化dp方程,如果j从0开始,只要初始化dp[0][0],如果从1开始,则dp[i][0]=1,0<=i<=n;
代码如下:
#include<bits/stdc++.h>
typedef long long ll;
ll f[102][102];
const ll mod=1e6+7;
std::vector<ll> a(103,0);
void solve()
{
ll n,m;
std::cin>>n>>m;
for(ll i=1;i<=n;i++)
std::cin>>a[i];
for(ll i=0;i<=n;i++)f[i][0]=1;
for(ll i=1;i<=n;i++)
for(ll j=1;j<=m;j++)
for(ll k=0;k<=std::min(a[i],j);k++)
f[i][j]=(f[i][j]%mod+f[i-1][j-k]%mod)%mod;
std::cout<<f[n][m];
}
int main()
{
std::ios_base::sync_with_stdio(false);
std::cin.tie(0);
std::cout.tie(0);
solve();
return 0;
}
一维dp方法:
1.花盆种类正向遍历,花盆总容量倒向遍历,m>=0,并且k从1开始(一维dp开始dp[n]=dp[n-1],不需要考虑0).
代码如下:
#include<bits/stdc++.h>
typedef long long ll;
ll f[102];
const ll mod=1e6+7;
std::vector<ll> a(103,0);
void solve()
{
ll n,m;
std::cin>>n>>m;
for(ll i=1;i<=n;i++)
std::cin>>a[i];
f[0]=1;
for(ll i=1;i<=n;i++)
for(ll j=m;j>=0;j--)
for(ll k=1;k<=std::min(a[i],j);k++)
f[j]=(f[j]%mod+f[j-k]%mod)%mod;
std::cout<<f[m];
}
int main()
{
std::ios_base::sync_with_stdio(false);
std::cin.tie(0);
std::cout.tie(0);
solve();
return 0;
}
谢谢观看