題目
一個有名的按摩師會收到源源不斷的預約請求,每個預約都可以選擇接或不接。在每次預約服務之間要有休息時間,因此她不能接受相鄰的預約。給定一個預約請求序列,替按摩師找到最優的預約集合(總預約時間最長),返回總的分鐘數。
示例:
輸入:[1, 4, 2, 4, 2, 3, 2, 5]
輸出:16
解釋:[4,4,3,5] 相加等于16
輸入: [2,1,4,5,3,1,1,3]
輸出: 12
解釋: 選擇 1 號預約、 3 號預約、 5 號預約和 8 號預約,總時長 = 2 + 4 + 3 + 3 = 12
解答
1、
public static int massage(int[] nums) {
int n = nums.length;
if (n == 0) {
return 0;
}
if (n == 1) {
return nums[0];
}
int[] result = new int[n];
result[0] = nums[0];
result[1] = Math.max(nums[0], nums[1]);
//將最大元素相加的值存入最后一個下標
for (int i = 2; i < n; i++) {
result[i] = Math.max(result[i - 1], result[i - 2] + nums[i]);
}
return result[n - 1];
}
2、
public static int massage2(int[] nums) {
int a = 0, b = 0;
for (int i = 0; i < nums.length; i++) {
int max = Math.max(a, b + nums[i]);
// 1 = Math(0,0+1)
// 4 = Math(1,0 + 4);
// 4 = Math(4,1+2);
b = a;
// 0 = 0
// b = 1
// b = 4
a = max;
// a = 1
// a = 4
// a = 8
}
return a;
}
測試類
public static void main(String[] args) {
int[] ints = {1, 4, 2, 4, 2, 3, 2, 5};
System.out.println(massage2(ints));
}
分析
時間復雜度 O(n), 空間復雜度O(1)