leetcode：按摩师(java)

题目

思路
立刻就想到动态规划了
先写出简单的二维动态规划的状态转移方程

状态转移方程
有两种选择，选或不选，用1，0表示
dp[i][0] 表示第i个预约不服务的最大预约时长
dp[i][1]表示第i个预约服务的最大预约时长
dp[i][0]的前一预约可以服务，也可以不服务，因此dp[i][0]取dp[i-1][0]和dp[i-1][1]的最大值

dp[i][0] = Math.max(dp[i-1][0],dp[i-1][1]);

dp[i][1]的前一个预约一定不服务，所以dp[i][1]等于前一个预约不服务的最长预约时长+第i个预约的时长

dp[i][1] = dp[i-1][0]+nums[i];

初始化
第一个预约不服务
dp[0][0] = 0;
第一个预约服务，最长时长就是第一个预约的时长
dp[0][1] = nums[0];

具体代码

class Solution {
    public int massage(int[] nums) {
        if(nums == null || nums.length == 0){
            return 0;
        }
        int len = nums.length;
        int[][] dp = new int[len][2];
        dp[0][0] = 0;
        dp[0][1] = nums[0];
        for(int i = 1;i < len;i++){
            dp[i][0] = Math.max(dp[i-1][0],dp[i-1][1]);
            dp[i][1] = dp[i-1][0]+nums[i];
        }
        return Math.max(dp[len-1][0],dp[len-1][1]);
    }
}

一维优化
dp[i]表示到第i个预约时的最长预约时间。
若第i个预约不进行服务，那么第i-1个预约是否服务无所谓，此时到第i个预约的最长预约时间为到第i-1个预约的最长预约时间，即dp[i-1]
若第i个预约要进行服务，那么第i-1个预约必须不进行服务，那么到第i-1个预约的最长预约时间也即到第i-2个预约的最长预约时间(第i-2个预约是否服务无所谓)，此时到第i-1个预约的最长预约时间为dp[i-2]+nums[i]
因此到第i个预约的最长预约时间为：

dp[i] = Math.max(dp[i-1],dp[i-2]+nums[i]);

具体代码

class Solution {
    public int massage(int[] nums) {
        if(nums == null || nums.length == 0){
            return 0;
        }
        if(nums.length == 1){
            return nums[0];
        }
        int len = nums.length;
        int[] dp = new int[len];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0],nums[1]);
        for(int i = 2;i < len;i++){
           dp[i] = Math.max(dp[i-1],dp[i-2]+nums[i]);
        }
        return dp[len - 1];
    }
}