题目链接
Leetcode.2070 每一个查询的最大美丽值
rating : 1724
题目描述
给你一个二维整数数组 i t e m s items items,其中 i t e m s [ i ] = [ p r i c e i , b e a u t y i ] items[i] = [pricei, beautyi] items[i]=[pricei,beautyi] 分别表示每一个物品的 价格 和 美丽值 。
同时给你一个下标从 0 0 0 开始的整数数组 q u e r i e s queries queries 。对于每个查询 q u e r i e s [ j ] queries[j] queries[j],你想求出价格小于等于 q u e r i e s [ j ] queries[j] queries[j] 的物品中,最大的美丽值 是多少。如果不存在符合条件的物品,那么查询的结果为 0 0 0。
请你返回一个长度与 q u e r i e s queries queries 相同的数组 a n s w e r answer answer ,其中 a n s w e r [ j ] answer[j] answer[j] 是第 j 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 ≤ i t e m s . l e n g t h , q u e r i e s . l e n g t h ≤ 1 0 5 1 \leq items.length, queries.length \leq 10^5 1≤items.length,queries.length≤105
- i t e m s [ i ] . l e n g t h = 2 items[i].length = 2 items[i].length=2
- 1 ≤ p r i c e i , b e a u t y i , q u e r i e s [ j ] ≤ 1 0 9 1 \leq pricei, beautyi, queries[j] \leq 10^9 1≤pricei,beautyi,queries[j]≤109
解法:排序 + 二分
首先我们先对 i t e m s items items 按价格 p r i c e price price 从小到大排序。
排序前,[ [1,2] , [3,2] , [2,4] , [5,6] ]
。
排序后,[ [1,2] , [2,4] , [3,2] , [5,6] ]
。
如果我们此时给定一个
q
=
3
q = 3
q=3 ,求 价格小于等于
q
q
q 的最大美丽值。第二个,即
[
2
,
4
]
[2,4]
[2,4] 就是我们要选择的那个,最大美丽值是 4
。
实际上我们只需要求 小于等于 q
的最大美丽值是多少,我们并不关心具体要选择哪一个。
所以我们可以直接让排序后的
i
t
e
m
s
items
items 的
b
e
a
u
t
y
i
beauty_i
beautyi 都是到目前为止的最大值。即 [ [1,2] , [2,4] , [3,4] , [5,6] ]
。
这下我们可以对于任何一个 queries[i]
通过 二分 的方式求它的最大美丽值。
时间复杂度: O ( n ∗ l o g ( q u e r i e s . l e n g t h ) ) O(n * log(queries.length)) O(n∗log(queries.length))
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;
}
}