1. 问题描述:
小明的花店新开张,为了吸引顾客,他想在花店的门口摆上一排花,共m 盆。通过调查顾客的喜好,小明列出了顾客最喜欢的n 种花,从1 到n 标号。为了在门口展出更多种花,规定第i 种花不能超过ai 盆,摆花时同一种花放在一起,且不同种类的花需按标号的从小到大的顺序依次摆列。试编程计算,一共有多少种不同的摆花方案。
输入描述:
第一行包含两个正整数n和m,中间用一个空格隔开。
第二行有n个整数,每两个整数之间用一个空格隔开,依次表示a1、a2、……an。
输出描述:
输出只有一行,一个整数,表示有多少种方案。注意:因为方案数可能很多,请输出方案数对1000007取模的结果。
示例1
输入
2 4
3 2
输出
2
说明
有2种摆花的方案,分别是(1,1,1,2),(1,1,2,2)。括号里的1和2表示两种花,比如第一个方案是前三个位置摆第一种花,第四个位置摆第二种花。
备注:
对于20%数据,有0<n≤8,0<m≤8,0≤ai≤8;
对于50%数据,有0<n≤20,0<m≤20,0≤ai≤20;
对于100%数据,有0<n≤100,0<m≤100,0≤ ai≤100。
2. 思路分析:
这道题目是一道经典的动态规划题目,分析题目可以知道存在两个动态变化的参数(一般都是需要结合题目分析出具有多少个动态变化的参数这样可以确定大概可以声明多少维的dp),分别是花的种类与花的盆数,所以基于这个想法我们可以声明一个二维dp列表(python语言为列表,其他语言例如c++/java可以声明数组)看怎么样根据题目的意思确定dp列表每一维的含义。我们知道放的花都是由编号小的到编号大的所以第一维可以表示当前放的是第i种花,因为花的盆数是不可以超过m盆所以我们可以这样想,到当前的第i种花为止放的花的盆数为j,所以第二维可以表示当前1~i种花放的盆数。当二维列表的含义确定之后剩下来的就比较好办了,状态转移也很好想,由上一种i-1花的盆数转移到当前第i种花的盆目,使用两层循环递推即可。
3. 代码如下:
if __name__ == '__main__':
n, m = map(int, input().split())
# a列表存储的是每一种喜欢的花的数量限制
a = list(map(int, input().split()))
# 插入一个元素可以使下标从1开始
a.insert(0, 0)
dp = [[0] * 1000 for i in range(1000)]
dp[0][0] = 1
mod = 1000007
# 第i种花
for i in range(1, n + 1):
# 到当前的第i种花放的数目总共有j盆
for j in range(m + 1):
# 当前的第i种花最多可以放k盆, 第i-1种最多可以放j-k
for k in range(a[i] + 1):
if j - k >= 0:
dp[i][j] = (dp[i][j] + dp[i - 1][j - k]) % mod
# 当前的第i种花已经放不下了
else:break
print(dp[n][m])