Leetcode.2070 每一个查询的最大美丽值

感觉画质不如…原神

已于 2023-12-03 16:20:29 修改

阅读量171

点赞数 2

分类专栏： Leetcode 文章标签： leetcode 算法二分

于 2023-03-08 13:08:15 首次发布

本文链接：https://blog.csdn.net/m0_74396439/article/details/129401140

版权

Leetcode 专栏收录该内容

441 篇文章 2 订阅

订阅专栏

题目链接

Leetcode.2070 每一个查询的最大美丽值 rating ： 1724

题目描述

给你一个二维整数数组 $i t e m s$ ，其中 $i t e m s [i] = [p r i ce i, b e a u t y i]$ 分别表示每一个物品的价格和 美丽值 。

同时给你一个下标从 $0$ 开始的整数数组 $q u er i es$ 。对于每个查询 $q u er i es [j]$ ，你想求出价格小于等于 $q u er i es [j]$ 的物品中，最大的美丽值 是多少。如果不存在符合条件的物品，那么查询的结果为 $0$ 。

请你返回一个长度与 $q u er i es$ 相同的数组 $an s w er$ ，其中 $an s w er [j]$ 是第 $j$ 个查询的答案。

示例 1：

输入：items = [[1,2],[3,2],[2,4],[5,6],[3,5]], queries = [1,2,3,4,5,6]
输出：[2,4,5,5,6,6]
解释：
queries[0]=1 ，[1,2] 是唯一价格 <= 1 的物品。所以这个查询的答案为 2 。
queries[1]=2 ，符合条件的物品有 [1,2] 和 [2,4] 。
它们中的最大美丽值为 4 。
queries[2]=3 和 queries[3]=4 ，符合条件的物品都为 [1,2] ，[3,2] ，[2,4] 和 [3,5] 。
它们中的最大美丽值为 5 。
queries[4]=5 和 queries[5]=6 ，所有物品都符合条件。
所以，答案为所有物品中的最大美丽值，为 6 。

示例 2：

输入：items = [[1,2],[1,2],[1,3],[1,4]], queries = [1]
输出：[4]
解释：
每个物品的价格均为 1 ，所以我们选择最大美丽值 4 。
注意，多个物品可能有相同的价格和美丽值。

示例 3：

输入：items = [[10,1000]], queries = [5]
输出：[0]
解释：
没有物品的价格小于等于 5 ，所以没有物品可以选择。
因此，查询的结果为 0 。

提示：

$\leq items.length, queries.length \leq 10^5$
$i t e m s [i] . l e n g t h = 2$
$\leq pricei, beautyi, queries[j] \leq 10^9$

解法：排序 + 二分

首先我们先对 $i t e m s$ 按价格 $p r i ce$ 从小到大排序。

排序前，[ [1,2] , [3,2] , [2,4] , [5,6] ]。

排序后，[ [1,2] , [2,4] , [3,2] , [5,6] ]。

如果我们此时给定一个 $q = 3$ ，求价格小于等于 $q$ 的最大美丽值。第二个，即 $[2, 4]$ 就是我们要选择的那个，最大美丽值是 4。

实际上我们只需要求小于等于 q的最大美丽值是多少，我们并不关心具体要选择哪一个。

所以我们可以直接让排序后的 $i t e m s$ 的 $beauty_i$ 都是到目前为止的最大值。即 [ [1,2] , [2,4] , [3,4] , [5,6] ]。

这下我们可以对于任何一个 queries[i]通过二分的方式求它的最大美丽值。

时间复杂度： $O (n * l o g (q u er i es . l e n g t h))$

C++代码：

class Solution {
public:
    vector<int> maximumBeauty(vector<vector<int>>& items, vector<int>& queries) {
        sort(items.begin(),items.end(),[&](auto &a,auto &b){
            return a[0] < b[0];
        });

        int n = items.size();
        for(int i = 1;i < n;i++){
            items[i][1] = max(items[i][1],items[i-1][1]);
        }

        vector<int> ans(queries.size());
        for(int i = 0;i <queries.size();i++){
            int q = queries[i];

            int l = 0,r = n - 1;
            while(l < r){
                int mid = (l + r) >> 1;
                if(items[mid][0] > q) r = mid;
                else l = mid + 1;
            }

            if(items[l][0] > q && l > 0) ans[i] = items[l-1][1];
            else if(items[l][0] <= q) ans[i] = items[l][1];
        }
        return ans;
    }
};

Java代码：

class Solution {
    public int[] maximumBeauty(int[][] items, int[] queries) {
        Arrays.sort(items,(a,b)->(a[0] - b[0]));

        for(int i = 1;i < items.length;i++){
            items[i][1] = Math.max(items[i][1],items[i-1][1]);
        }

        int n = queries.length;
        int[] ans = new int[n];

        for(int i = 0;i < n;i++){
            int q = queries[i];

            int l = 0;
            int r = items.length - 1;

            while(l < r){
                int mid = (l + r) >> 1;
                if(items[mid][0] > q) r = mid;
                else l = mid + 1;
            }

            if(items[l][0] > q && l > 0) ans[i] = items[l-1][1];
            else if(items[l][0] <= q) ans[i] = items[l][1];
        }
        return ans;
    }
}