一个有名的按摩师会收到源源不断的预约请求,每个预约都可以选择接或不接。在每次预约服务之间要有休息时间,因此她不能接受相邻的预约。给定一个预约请求序列,替按摩师找到最优的预约集合(总预约时间最长),返回总的分钟数。
注意:本题相对原题稍作改动
示例 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。
解题思路:
对于预约请求,从前往后处理,后一个状态依赖于前一个状态,因此采用动态规划的方法。
在此题中有忙和闲两个状态,当前时刻的忙只能是因为上一时刻闲,即 int tdp1 = dp0 + nums[i];而上一时刻的忙和闲都会导致该时刻闲,即 int tdp0 = Math.max(dp0,dp1);
代码实现:
import java.util.*;
public class Helloworld {
public int massage(int[] nums) {
int n = nums.length;
if (n == 0)
return 0;
int dp0 = 0, dp1 = nums[0];
for (int i = 1; i < n; i++) {
int tdp0 = Math.max(dp0,dp1);
int tdp1 = dp0 + nums[i];
dp0 = tdp0;
dp1 = tdp1;
}
return Math.max(dp0,dp1);
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
String string = scanner.nextLine();
String[] str = string.split(" ");
int[] nums = new int[str.length];
for (int i = 0; i < str.length; i++) {
nums[i] = Integer.parseInt(str[i]);
}
System.out.println(new Helloworld().massage(nums));
}
}