Java实现LeetCode 5451. 满足不等式的最大值 (类似先序队列)

 

给你一个数组 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
解释：前两个点满足 |xi - xj| <= 1 ，带入方程计算，则得到值 3 + 0 + |1 - 2| = 4 。第三个和第四个点也满足条件，得到值 10 + -10 + |5 - 6| = 1 。
没有其他满足条件的点，所以返回 4 和 1 中最大的那个。

示例 2：

输入：points = [[0,0],[3,0],[9,2]], k = 3
输出：3
解释：只有前两个点满足 |xi - xj| <= 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 是严格递增的。

import java.util.*;
class Solution {
    /*
        改题，yi + yj + |xi - xj| -->  yi + yj + |xj - xi| -->  yj + xj + (yi  - xi)
        对于每个j而言，只需要找到其前面k个点以内的 (yi  - xi) 最大值
        改成这样就明白了，可以用先序队列，（我之前用的），
        但是x一定是递增的，所以不一定要求先序队列，
        这里就用普通得集合就行，但对于效率得考虑，由于经常要修改集合，优先使用LinkedList
        ArrayList查询快，修改慢
    */
     public int findMaxValueOfEquation(int[][] points, int k) {
        Deque<Integer> q = new LinkedList<>();
        //先把第一个元素放进去，这里放的是下标
        q.add(0);
        
        int max =Integer.MIN_VALUE;
        //循环每一个点
        for(int j = 1;j<points.length;j++){
            //如果当前点和以前得点差大于k就要删掉得，因为x是递增的，所以集合里前面得x是整个集合最小的
            while(!q.isEmpty()&&points[q.peek()][0]+k<points[j][0]) q.removeFirst();
            //如果当前集合空了，就要把我当前得放入集合，，就可以去找下一个值了，
            //因为集合里没值，无法实现取值操作
            if(q.isEmpty()){
                q.addLast(j);
                continue;
            }
            //取出x最小得那个值得下标
            int index = q.peek();
            //去除当前这个公式得值  yj + xj + (yi  - xi)
            int cur = points[j][0]+points[j][1] + points[index][1] - points[index][0];
            //然后看集合里面最尾得值，看看(yi  - xi)是不是比当前这个小，如果比当前得小，就直接删除把
            //因为下一个值肯定离当前值近，里队列里得尾值远，

            while(!q.isEmpty()&&(points[q.getLast()][1]-points[q.getLast()][0])<(points[j][1]-points[j][0])) q.removeLast();
            //把当前得值放进队列
            q.addLast(j);
            //每次比较最大值
            if(max<cur) max = cur; 
        }
        return max;
    }
}