LeetCode 每日一题 1014. 最佳观光组合

题目描述

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/best-sightseeing-pair
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

My Solution

(A_i+A_j+i-j = [A_i+ i ]+ [A_j-j]) ，显然，对于 i , (A_i+i) 是定值， 只需要最大化 (A_j-j) 即可。
综上，反向遍历一次即可。复杂度 (O(n))

#pragma G++ optimize("O2")

class Solution {
 public:
  int maxScoreSightseeingPair(vector<int>& A) {
    const int n = A.size();
    int ans(INT_MIN), q(-1);
    for(int i = n - 1; i >= 0; --i) {
      if(q != -1 && A[i] + i + A[q] - q > ans) {
        ans = (A[i] + i + A[q] - q);
      }
      if(q == -1 || A[i] - i >= A[q] - q)
        q = i;
    }
    return ans;
  }
};