摆花，动态规划

题目描述

小明的花店新开张，为了吸引顾客，他想在花店的门口摆上一排花，共 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;
}

谢谢观看