摆花(动态规划)

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])

 

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值