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

168 阅读1分钟

这是我参与11月更文挑战的第14天,活动详情查看:2021最后一次更文挑战

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

给你一个二维整数数组 items ,其中 items[i] = [pricei, beautyi] 分别表示每一个物品的 价格 和 美丽值 。

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

请你返回一个长度与 queries 相同的数组 answer,其中 answer[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

提示:

  • 1 <= items.length, queries.length <= 10510^5
  • items[i].length == 2
  • 1 <= pricei, beautyi, queries[j] <= 10910^9

解题思路

  1. 对items数组按照价格大小进行排序
  2. 维护前缀最大美丽值数组,b[i]代表在当前下标i之前,出现的最大美丽值
  3. 对于每一个查询,使用二分查找,找出 小于等于 queries[j]中具有最大价格物品,根据前缀最大美丽值数组,可以得出小于等于 queries[j]物品中的最大美丽值

代码

class Solution {
public:
    vector<int> maximumBeauty(vector<vector<int>>& items, vector<int>& queries) {
        sort(items.begin(),items.end());
        int m=items[0][1];
        int n=items.size();
        vector<int> b(n);
        vector<int> a(n);
        for (int i = 0; i < n; ++i) {
            m=max(m,items[i][1]);
            b[i]=m;
            a[i]=items[i][0];
        }
        vector<int> res(queries.size());
        for (int i = 0; i < queries.size(); ++i) {
            int idx=upper_bound(a.begin(),a.end(),queries[i])-a.begin();
            if (idx==0)
                res[i]=0;
            else res[i]=b[idx-1];
        }
        return res;
    }

};