LeetCode-最大和查询

2736. 最大和查询

给你两个长度为 n 、下标从 0 开始的整数数组 nums1 和 nums2 ，另给你一个下标从 1 开始的二维数组 queries ，其中 queries[i] = [xi, yi] 。

对于第 i 个查询，在所有满足 nums1[j] >= xi 且 nums2[j] >= yi 的下标 j (0 <= j < n) 中，找出 nums1[j] + nums2[j] 的 最大值 ，如果不存在满足条件的 j 则返回 -1 。

返回数组 answer ，其中 answer[i] 是第 i 个查询的答案。

示例 1：

输入：nums1 = [4,3,1,2], nums2 = [2,4,9,5], queries = [[4,1],[1,3],[2,5]]
输出：[6,10,7]
解释：
对于第 1 个查询：xi = 4 且 yi = 1 ，可以选择下标 j = 0 ，此时 nums1[j] >= 4 且 nums2[j] >= 1 。nums1[j] + nums2[j] 等于 6 ，可以证明 6 是可以获得的最大值。
对于第 2 个查询：xi = 1 且 yi = 3 ，可以选择下标 j = 2 ，此时 nums1[j] >= 1 且 nums2[j] >= 3 。nums1[j] + nums2[j] 等于 10 ，可以证明 10 是可以获得的最大值。
对于第 3 个查询：xi = 2 且 yi = 5 ，可以选择下标 j = 3 ，此时 nums1[j] >= 2 且 nums2[j] >= 5 。nums1[j] + nums2[j] 等于 7 ，可以证明 7 是可以获得的最大值。
因此，我们返回 [6,10,7] 。

示例 2：

输入：nums1 = [3,2,5], nums2 = [2,3,4], queries = [[4,4],[3,2],[1,1]]
输出：[9,9,9]
解释：对于这个示例，我们可以选择下标 j = 2 ，该下标可以满足每个查询的限制。

示例 3：

输入：nums1 = [2,1], nums2 = [2,3], queries = [[3,3]]
输出：[-1]
解释：示例中的查询 xi = 3 且 yi = 3 。对于每个下标 j ，都只满足 nums1[j] < xi 或者 nums2[j] < yi 。因此，不存在答案。 

Java实现：

//暴力解法
class Solution {
    public int[] maximumSumQueries(int[] nums1, int[] nums2, int[][] queries) {
        int length = queries.length;
        int[] answer = new int[length];
        for(int i=0;i<length;i++){
            int itemMax=-1;
            for(int j=0;j<nums1.length;j++){
                if(nums1[j] >= queries[i][0]&&nums2[j] >= queries[i][1]){
                    int sum = nums1[j]+nums2[j];
                    if(itemMax<sum){
                        itemMax = sum;
                    }
                }
            }
            answer[i] = itemMax;
        }
        return answer;
    }
}

//离线排序 和 静态线段树反向查询
class Solution {
    public int[] maximumSumQueries(int[] nums1, int[] nums2, int[][] queries) {
        Integer[] numsSort = new Integer[nums1.length];
        Arrays.setAll(numsSort, (i) -> i);
        Arrays.sort(numsSort, (i1,i2) -> nums1[i1] - nums1[i2]);
        
        int[] ans = new int[queries.length];
        Integer[] ansSort = new Integer[queries.length];
        Arrays.setAll(ansSort, (i) -> i);
        Arrays.sort(ansSort, (i1,i2) -> queries[i1][0] - queries[i2][0]);

        int[] qPosX = new int[numsSort.length + 1];
        int[] qPosY = new int[numsSort.length + 1];
        int[] querie;
        int numsTail = numsSort.length - 1, qTail = 0, x, y, uper, lower, mider;
        qPosX[0] = Integer.MAX_VALUE;
        for (int i = ansSort.length - 1; i >= 0; i--) {
            querie = queries[ansSort[i]];

            while (numsTail >= 0 && nums1[numsSort[numsTail]] >= querie[0]) {
                if (qPosY[qTail] < nums2[numsSort[numsTail]]) {
                    x = nums1[numsSort[numsTail]];
                    y = nums2[numsSort[numsTail]];
                    while (x + y > qPosX[qTail] + qPosY[qTail]) qTail--;
                    qPosX[++qTail] = x;
                    qPosY[qTail] = y;
                }
                numsTail--;
            }

            // 二分
            // uper = qTail; lower = 1;
            // while (lower <= uper) {
            //     mider = (uper + lower) >> 1;
            //     if (qPosY[mider] < querie[1]) lower = mider + 1;
            //     else uper = mider - 1;
            // }
            // if (lower > qTail) ans[ansSort[i]] = -1;
            // else ans[ansSort[i]] = qPosX[uper + 1] + qPosY[uper + 1];

            // 枚举 - 高到低
            // uper = qTail;
            // while (qPosY[uper] >= querie[1]) uper--;
            // if (uper == qTail) ans[ansSort[i]] = -1;
            // else ans[ansSort[i]] = qPosX[uper + 1] + qPosY[uper + 1];

            // 枚举 - 低到高
            uper = 1;
            while (uper <= qTail && qPosY[uper] < querie[1]) uper++;
            ans[ansSort[i]] = uper > qTail ? -1 : qPosX[uper] + qPosY[uper];
        }

        return ans;
    }
}

LeetCode快速执行验证