题目
一个有名的按摩师会收到源源不断的预约请求,每个预约都可以选择接或不接。在每次预约服务之间要有休息时间,因此她不能接受相邻的预约。给定一个预约请求序列,替按摩师找到最优的预约集合(总预约时间最长),返回总的分钟数。注意:本题相对原题稍作改动。
示例 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。
链接: 力扣(LeetCode).
解题思路:
题目要求按摩预约不能连续,可以看成动态规划的最优解问题。题目要求出最优预约时间和,可以分解成子问题,dp[i]的最优解可以看作选和不选的问题,即选第i个,则dp[i]=dp[i-2]+nums[i],不选则dp[i]=dp[i-1]。dp[i]存储的是前面一个数dp[i-1]和往前第二个数dp[i-2]加上dp[i]两者之间的最大值。
动态方程:
dp[i] = max(dp[i-1],dp[i-2]+nums[i])
解题代码:
class Solution {
public int massage(int[] nums) {
//考虑到内存消耗,故不定义数组,也可解此题
int dp1=0,dp2=0,dp3=0;
//循环遍历最优解
for(int i = 0;i < nums.length;i++) {
//选择第i个和不选第i个,选择最优值
dp3 = Math.max(nums[i]+dp1,dp2);
dp1 = dp2;
dp2 = dp3;
}
return dp2;
}
}
代码效率:
注解:
此题和斐波那契数列及其相似,可参考其解法