面试题 17.16. 按摩师

本文旨在对于个人知识的梳理以及知识的分享，如果有不足的地方，欢迎大家在评论区指出

---

题目描述

一个有名的按摩师会收到源源不断的预约请求，每个预约都可以选择接或不接。在每次预约服务之间要有休息时间，因此她不能接受相邻的预约。给定一个预约请求序列，替按摩师找到最优的预约集合（总预约时间最长），返回总的分钟数。

注意：本题相对原题稍作改动
示例 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
}