给你一个二维整数数组 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 <= 105
items[i].length == 2
1 <= pricei, beautyi, queries[j] <= 109
方法一:排序 + 二分查找
class Solution:
def maximumBeauty(self, items: List[List[int]], queries: List[int]) -> List[int]:
items.sort()
prices = [p for p, _ in items]
mx = [items[0][1]]
for _, b in items[1:]:
mx.append(max(mx[-1], b))
ans = [0] * len(queries)
for i, q in enumerate(queries):
j = bisect_right(prices, q)
if j:
ans[i] = mx[j - 1]
return ans
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][1], items[i][1]);
}
int n = queries.length;
int[] ans = new int[n];
for (int i = 0; i < n; ++i) {
int left = 0, right = items.length;
while (left < right) {
int mid = (left + right) >> 1;
if (items[mid][0] > queries[i]) {
right = mid;
} else {
left = mid + 1;
}
}
if (left > 0) {
ans[i] = items[left - 1][1];
}
}
return ans;
}
}
class Solution {
public:
vector<int> maximumBeauty(vector<vector<int>>& items, vector<int>& queries) {
sort(items.begin(), items.end());
for (int i = 1; i < items.size(); ++i) items[i][1] = max(items[i - 1][1], items[i][1]);
int n = queries.size();
vector<int> ans(n);
for (int i = 0; i < n; ++i) {
int left = 0, right = items.size();
while (left < right) {
int mid = (left + right) >> 1;
if (items[mid][0] > queries[i])
right = mid;
else
left = mid + 1;
}
if (left) ans[i] = items[left - 1][1];
}
return ans;
}
};
func maximumBeauty(items [][]int, queries []int) []int {
sort.Slice(items, func(i, j int) bool {
return items[i][0] < items[j][0]
})
for i := 1; i < len(items); i++ {
items[i][1] = max(items[i-1][1], items[i][1])
}
n := len(queries)
ans := make([]int, n)
for i, v := range queries {
left, right := 0, len(items)
for left < right {
mid := (left + right) >> 1
if items[mid][0] > v {
right = mid
} else {
left = mid + 1
}
}
if left > 0 {
ans[i] = items[left-1][1]
}
}
return ans
}
func max(a, b int) int {
if a > b {
return a
}
return b
}