面试题 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。

---

解法一： 动态规划+贪心

• 建一个dp表存储到达该点的最长预约时间，并用一个res贪心存储表中最长时间

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];
        int res=Math.max(nums[0],nums[1]);
        dp[0]=nums[0];
        dp[1]=nums[1];
        for(int i=2;i<n;i++){
            int index=i-2;
            int max=dp[index--];
            while(index>=0){
                max=Math.max(max,dp[index--]);
            }
            dp[i]=nums[i]+max;
            res=Math.max(res,dp[i]);
        }
        return res;
    }
}

时间优化

• 状态方程：dp[i] = max(dp[i - 1], dp[i - 2] + nums[i])
• dp[i-1]表示：dp[0]-dp[i-3]的最长预约时间+nums[i-1];
• dp[i-2]+nums[i]：dp[0]-dp[i-2]最长预约时间+nums[i];
• 在两者之中选择最大值即总体最长。 (不需要从dp[0]开始扫描)

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[1],nums[0]);
        for(int i=2;i<n;i++){
            dp[i]=Math.max(dp[i-1],dp[i-2]+nums[i]);
        }
        return dp[n-1];
    }
}

空间优化

• 用两个指针代替dp表

class Solution {
    public int massage(int[] nums) {
        int n=nums.length;
        if(n==0) return 0;
        if(n==1) return nums[0];
        int pre=nums[0];
        int res=Math.max(pre,nums[1]);
        for(int i=2;i<n;i++){
            int temp=Math.max(res,pre+nums[i]);
            pre=res;
            res=temp;
        }
        return res;
    }
}

代码优化

class Solution {
    public int massage(int[] nums) {
        int n=nums.length;
        int pre=0;
        int res=0;
        //从头开始
        for(int i=0;i<n;i++){
            int temp=Math.max(res,pre+nums[i]);
            pre=res;
            res=temp;
        }
        return res;
    }
}