本文旨在对于个人知识的梳理以及知识的分享,如果有不足的地方,欢迎大家在评论区指出
题目描述
一个有名的按摩师会收到源源不断的预约请求,每个预约都可以选择接或不接。在每次预约服务之间要有休息时间,因此她不能接受相邻的预约。给定一个预约请求序列,替按摩师找到最优的预约集合(总预约时间最长),返回总的分钟数。
注意:本题相对原题稍作改动
示例 1:
输入: [1,2,3,1]
输出: 4
解释: 选择 1 号预约和 3 号预约,总时长 = 1 + 3 = 4。
示例 2:
输入: [2,7,9,3,1]
输出: 12
解释: 选择 1 号预约、 3 号预约和 5 号预约,总时长 = 2 + 9 + 1 = 12。
示例 3:
输入: [2,1,4,5,3,1,1,3]
输出: 12
解释: 选择 1 号预约、 3 号预约、 5 号预约和 8 号预约,总时长 = 2 + 4 + 3 + 3 = 12。
题目链接
题目分析
这道题是一道动态规划的问题,首先我们要想清楚状态的表示,在这里我们可以使用f[i]
表示前i个人的预约可以完成的最大分钟数,那么状态转移方程我们可以根据接不接受第i
个预约来划分,也就是f[i] = max(f[i-2]+nums[i], f[i-1])
解题代码
Go
func massage(nums []int) int {
n := len(nums)
if n == 0 {
return 0
}
if n == 1 {
return nums[0]
}
f := make([]int, n, n)
f[0] = nums[0]
f[1] = int(math.Max(float64(nums[0]), float64(nums[1])))
for i:=2; i<n; i++ {
f[i] = int(math.Max(float64(f[i-2]+nums[i]), float64(f[i-1])))
}
return f[n-1]
}
可以发现,当前的状态只与前两个状态相关,所以我们可以将空间优化为一维
func massage(nums []int) int {
n := len(nums)
a, b := 0, 0
for i:=0; i<n; i++ {
c := b
b = a
a = int(math.Max(float64(a), float64(c+nums[i])))
}
return a
}