当前位置：首页 > article >正文

leetcode做题笔记2736. 最大和查询

article 2025/3/1 7:27:44

给你两个长度为 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 。因此，不存在答案。

思路一：模拟题意

c++解法

class Solution {
public:
    vector<int> maximumSumQueries(vector<int>& nums1, vector<int>& nums2, vector<vector<int>>& queries) {
        int n = nums1.size();
        vector<pair<int, int>> v;
        for(int i = 0; i < nums1.size(); ++i) {
            v.emplace_back(nums1[i] + nums2[i], i);
        }
        map<pair<int, int>, int> map;
        sort(v.begin(), v.end(), [&](pair<int, int>& p1, pair<int, int>& p2){
            return p1.first > p2.first;
        });
        int m = queries.size();
        vector<int> ans(m, -1);
        for(int i = 0; i < m; ++i) {
            int q1 = queries[i][0], q2 = queries[i][1];
            if(map.count({q1, q2})) ans[i] = map[{q1, q2}];
            else {
                for(int j = 0; j < n; ++j) {
                    if(nums1[v[j].second] >= q1 && nums2[v[j].second] >= q2) {
                        ans[i] = v[j].first;
                        map[{q1, q2}] = v[j].first;
                        break;
                    }
                }
            }
        }
        return ans;
    }
};