面试题 17.16. 按摩师 Java

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

注意：本题相对原题稍作改动

示例 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。

Java(DP)

class Solution {
    public int massage(int[] nums) {
        int n = nums.length;
        if(n == 0){
            return 0;
        }
        if(n == 1){
            return nums[0];
        }
        int[] dp = new int[n];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0], nums[1]);
        for(int i = 2; i < n; i++){
            dp[i] = Math.max(dp[i-1], dp[i-2] + nums[i]);
            //最优解受上一次，和上上一次的影响
            //所以其实很简单，因为无非就两种情况。
            //这次接了单，相邻的又不能接，那总和就是上上一次的dp加上这次的num
            //这次没接单，相邻的也不能接，那总和不就和上次的dp一样了。
            //然后把这两种情况的max作为dp就行了。
        }
        return dp[n-1];
    }
}

动态规划

说白了动态规划的核心就是状态转移方程，这种题目一般能解出转移方程，那就几乎算是做完了，剩下能做的也就是考虑考虑边界条件，和补齐base case而已。所以可能动态规划考验的还是数学能力。能否从逻辑上，把具体场景的问题，转化成数学上的转移方程，这才是动态规划的关键。