leetcode算法学习(6)——最佳观光组合

题目描述

给定正整数数组 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

简单粗暴的解法(超出时间限制)

class Solution:
    def maxScoreSightseeingPair(self, A: List[int]) -> int:
        max_score=0
        for i in range(len(A)):
            for j in range(i+1,len(A)):
                score=A[i]+A[j]+i-j
                if score>=max_score:
                    max_score=score
        return max_score

改进思路

• 已知题目要求max_score=A[i] + A[j] + i - j （i < j） 的最大值，而对于输入中的每一个 A[j] 来说， 它的值 A[j] 和它的下标 j 都是固定的，所以 A[j] - j 的值也是固定的。
• 因此，对于每个 A[j] 而言， 想要求 max_score 的最大值，也就是要求 A[i] + i （i < j） 的最大值，
• 所以不妨用一个变量 pre_max 记录当前元素 A[j] 之前的 A[i] + i 的最大值，这样对于每个 A[j] 来说，都有最大得分 = pre_max + A[j] - j，再从所有 A[j] 的最大得分里挑出最大值返回即可。

代码

class Solution:
    def maxScoreSightseeingPair(self, A: List[int]) -> int:
        max_score=0
        pre_max=A[0]+0
        for j in range(1,len(A)):
            max_score=max(max_score,pre_max+A[j]-j)
            pre_max=max(pre_max,A[j]+j)
        return max_score

进一步优化时间

class Solution:
    def maxScoreSightseeingPair(self, A: List[int]) -> int:
        max_score=0
        pre_max=A[0]
        for j in range(1,len(A)):
            max_score=max_score if max_score>=pre_max+A[j]-j else pre_max+A[j]-j
            pre_max=pre_max if pre_max>=A[j]+j else A[j]+j
        return max_score