给定正整数数组 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
代码:
暴力双循环有测试用例会超时,所以需要优化一下。
这里可以把计算公式变化一下变成:
(A[j] - j) + (A[i] + i)
使用两个变量分别记录当前i景点的值和i景点+j景点的总值
这样经过一次遍历就可以获得答案
func maxScoreSightseeingPair(A []int) int {
lenth := len(A)
maxI := A[0]
ret := 0
for i := 1; i < lenth; i++ {
ret = max(maxI + A[i] - i, ret)
maxI = max(maxI, A[i] + i)
}
return ret
}
func max(a, b int) int {
if a > b {
return a
}
return b
}