[LeetCode解题报告] 1499. 满足不等式的最大值

一、 题目

1. 题目描述

给你一个数组 points 和一个整数 k 。数组中每个元素都表示二维平面上的点的坐标，并按照横坐标 x 的值从小到大排序。也就是说 points[i] = [xi, yi] ，并且在 1 <= i < j <= points.length 的前提下， xi < xj 总成立。

请你找出 yi + yj + |xi - xj| 的 最大值，其中 |xi - xj| <= k 且 1 <= i < j <= points.length。

题目测试数据保证至少存在一对能够满足 |xi - xj| <= k 的点。

 

示例 1：

输入：points = [[1,3],[2,0],[5,10],[6,-10]], k = 1
输出：4
解释：前两个点满足 |x_i - x_j| <= 1 ，代入方程计算，则得到值 3 + 0 + |1 - 2| = 4 。第三个和第四个点也满足条件，得到值 10 + -10 + |5 - 6| = 1 。
没有其他满足条件的点，所以返回 4 和 1 中最大的那个。

示例 2：

输入：points = [[0,0],[3,0],[9,2]], k = 3
输出：3
解释：只有前两个点满足 |x_i - x_j| <= 3 ，代入方程后得到值 0 + 0 + |0 - 3| = 3 。

 

提示：

• 2 <= points.length <= 10^5
• points[i].length == 2
• -10^8 <= points[i][0], points[i][1] <= 10^8
• 0 <= k <= 2 * 10^8
• 对于所有的1 <= i < j <= points.length ，points[i][0] < points[j][0] 都成立。也就是说，xi 是严格递增的。

Related Topics

• 队列
• 数组
• 滑动窗口
• 单调队列
• 堆（优先队列）

• 👍 55
• 👎 0

2. 原题链接

链接: 1499. 满足不等式的最大值

二、 解题报告

1. 思路分析

单调队列，令j>i,则yi + yj + |xi - xj|转化为(yi-xi)+(yj+xj)，
问题转化为对每个j来说，求左边的i里，(yi-xi)最大的数，最后求和统计max即可。
由于x[i]递增，所以可以维护一个单调递降队列，
证明:对区间[i,j]来说，若k是贡献(yk-xk)最大的数,那么所有k左边的数都不会对j后边的数产生贡献。

使用队列时，先用k移除队首太远的数，添加队尾时，为了维护单调性，需要while移除所有小于y-x的数

2. 复杂度分析

单调队列的时间复杂度理论上平均是O(1)
因此总体时间复杂度O(n)

3. 代码实现

单调队列维护下标

class Solution:
    def findMaxValueOfEquation(self, points: List[List[int]], k: int) -> int:        
        """
        x是有序的，因此当i<j原式可转化为
             yi+yj+xj-xi
            =(yj+xj+(yi-xi)
        实际是找数对(i,j)使上式最大。
        那么当我们遍历j时，只需要知道前边i中最大的(yi-xi)，且满足xj-xi<=k。
        若我们不看k条件，那么就可以直接遍历时同时维护一个max{yi-xi}即可。
        加上k条件看，由于x递增，因此实际是求j左边一个连续区间的max{yi-xi}，
        这个区间随着j右移而右移，区间左端点由k条件右移，这是单调队列擅长的。
        我们用单调队列维护这个区间的下标,
        因为要找数对，i!=j,因此先出队，处理，再入队更新。
        遍历j时，先移除过远的(xi+k<xj)队首，然后用剩下的队首更新答案，注意这里可能队列为空，则不更新。
        入队时，移除队尾所有y-x小于本j的数据，他们对后续的贡献不会超过本j。
        """
        ans = -1e9
        q = deque([0])
        for j in range(1,len(points)):
            x,y = points[j]
            while q and points[q[0]][0] + k < x:
                q.popleft()
            if q:
                xi,yi = points[q[0]]
                ans = max(ans,x+y+yi-xi)
            while q and points[q[-1]][1] - points[q[-1]][0] <= y-x:
                q.pop()
            q.append(j)
        
        return ans

单调队列维护i值

class Solution:
    def findMaxValueOfEquation(self, points: List[List[int]], k: int) -> int:
        ans = -1e13
        q = deque()
        x,y = points[0]
        q.append((y-x,x))
        for i in range(1,len(points)):
            x,y = points[i]
            while q and x - q[0][1] > k:
                q.popleft()

            if q:
                ans = max(ans,q[0][0] + x + y)   
            while q and q[-1][0] <= y-x:                        
                q.pop()              
                
            q.append((y-x,x))
   
        return ans

三、 本题小结

1. 求数组中两侧最大/最小值的题，经常可以用到单调队列/单调栈。