最佳观光组合
题目描述
给定正整数数组 A
,A[i]
表示第 i
个观光景点的评分,并且两个景点 i
和 j
之间的距离为 j - i
。
一对景点(i < j)
组成的观光组合的得分为(A[i] + A[j] + i - j)
:景点的评分之和减去它们两者之间的距离。
返回一对观光景点能取得的最高分。
示例:
输入:[8,1,5,2,6]
输出:11
解释:i = 0, j = 2, A[i] + A[j] + i - j = 8 + 5 + 0 - 2 = 11
提示:
2 <= A.length <= 50000
1 <= A[i] <= 1000
分析:
最简单的是使用枚举,很容易看出时间复杂度O(N^2)
,一看有50000的规模,肯定超时。带着侥幸的心理写完代码,一提交流下了不学无术的眼泪遇到题目还是先分析比较好。
下面分析一下这里的目标函数:A[i] + A[j] + i - j
,首先我们进行分解:
A
[
i
]
+
A
[
j
]
+
i
−
j
=
A
[
i
]
+
i
+
A
[
j
]
−
j
,
i
<
j
A[i] + A[j] + i -j = A[i]+i+A[j] - j \\\qquad \qquad \qquad ,i < j
A[i]+A[j]+i−j=A[i]+i+A[j]−j,i<j
将其拆分成 A[i] + i
和 A[j] - j
两部分。最大化目标函数,其实就等价于求中 A[i] + i
的最大值 tmx
,景点 j 的答案即为 res = tmx + A[j] - j
代码:
class Solution {
public int maxScoreSightseeingPair(int[] A) {
if (A == null || A.length < 2){
return 0;
}
int tmx = A[0] + 0;
int res = 0;
for (int i = 1; i < A.length; i++) {
res = Math.max(res, tmx + A[i] - i);
tmx = Math.max(tmx, A[i] + i);
}
return res;
}
}