动态规划模板题：采药

题目链接

采药

题目大意

辰辰是个天资聪颖的孩子，他的梦想是成为世界上最伟大的医师。为此，他想拜附近最有威望的医师为师。医师为了判断他的资质，给他出了一个难题。医师把他带到一个到处都是草药的山洞里对他说：“孩子，这个山洞里有一些不同的草药，采每一株都需要一些时间，每一株也有它自身的价值。我会给你一段时间，在这段时间里，你可以采到一些草药。如果你是一个聪明的孩子，你应该可以让采到的草药的总价值最大。”

如果你是辰辰，你能完成这个任务吗？

输入格式

第一行有 $2$ 个整数 $T$（$1\le T\le 1000$）和 $M$（$1\le M\le 100$），用一个空格隔开，$T$ 代表总共能够用来采药的时间，$M$ 代表山洞里的草药的数目。

接下来的 $M$ 行每行包括两个在 $1$ 到 $100$ 之间（包括 $1$ 和 $100$）的整数，分别表示采摘某株草药的时间和这株草药的价值。

输出格式

输出在规定的时间内可以采到的草药的最大总价值。

样例输入

70 3
71 100
69 1
1 2

样例输出

3

数据范围

• $1\le T\le 1000$
• $1\le M\le 100$

运行限制

• 最大运行时间：1s
• 最大运行内存: 128M

解题思路

动态规划

$0/1$背包问题（换汤不换药）

总的采药时间 $T$ $⟺$背包的总容量；
每株草药的采集时间$⟺$装入背包的每个物品所占体积；
草药价值$⟺$物品价值。

模板步骤：

1、定义两个长度为 $M+1$ 的列表 htime 和 hvalue 存储每个草药的采集时间和价值；
2、定义动态规划数组 dp$[M+1][T+1]$ ，dp$[i][j]$ 表示总采集时间为 $j$ 时，前 $i$ 个草药所能获得的最大价值；
3、其状态转移方程如下：
$$\begin{gathered} \mathrm{dp}\left[i\right]\left[j\right]=\{\begin{array}{l}\mathrm{dp}\left[i-1\right]\left[j\right]\mathrm{htime}\left[i\right]>j\\ \max \left(\mathrm{dp}\left[i-1\right]\left[j\right], \\ \mathrm{dp}\left[i-1\right]\left[j-\mathrm{htime}\left[i\right]\right]+\mathrm{hvalue}\left[i\right]\right)\mathrm{htime}\left[i\right]⩽j\end{array} \end{gathered}$$

4、最后的 dp$[M][T]$ 即为结果。

AC Code

Python3

不用滚动数组

# -*- coding: utf-8 -*-
# @Author : BYW-yuwei
# @Software: python3.8.6

T,M=map(int,input().split())
htime=[0]*(M+1)
hvalue=[0]*(M+1)
for i in range(1,M+1):
    htime[i],hvalue[i]=map(int,input().split())

dp=[[0]*(T+1) for i in range(M+1)]
for i in range(1,M+1):
    for j in range(0,T+1):
        if htime[i]>j:
            dp[i][j]=dp[i-1][j]
        else:
            npick=dp[i-1][j]
            pick=dp[i-1][j-htime[i]]+hvalue[i]
            dp[i][j]=max(npick,pick)
print(dp[M][T])

交替滚动

# -*- coding: utf-8 -*-
# @Author : BYW-yuwei
# @Software: python3.8.6
T,M=map(int,input().split())
htime=[0]*(M+1)
hvalue=[0]*(M+1)
for i in range(1,M+1):
    htime[i],hvalue[i]=map(int,input().split())
dp=[[0]*(T+1) for i in range(2)]
new = 0
old = 1
for i in range(1,M+1):
    new,old = old,new
    for j in range(0,T+1):
        if htime[i]>j:
            dp[new][j]=dp[old][j]
        else:
            npick=dp[old][j]
            pick=dp[old][j-htime[i]]+hvalue[i]
            dp[new][j]=max(npick,pick)
print(dp[new][T])

自身滚动

# -*- coding: utf-8 -*-
# @Author : BYW-yuwei
# @Software: python3.8.6
T,M=map(int,input().split())
htime=[0]*(M+1)
hvalue=[0]*(M+1)
for i in range(1,M+1):
    htime[i],hvalue[i]=map(int,input().split())
dp=[0]*(T+1)
for i in range(1,M+1):
    for j in range(T,htime[i]-1,-1):
        npick=dp[j]
        pick=dp[j-htime[i]]+hvalue[i]
        dp[j]=max(npick,pick)
print(dp[T])