【1333. 餐厅过滤器】
来源：力扣（LeetCode）
描述：
给你一个餐馆信息数组 restaurants，其中 restaurants[i] = [idi, ratingi, veganFriendlyi, pricei, distancei]。你必须使用以下三个过滤器来过滤这些餐馆信息。
其中素食者友好过滤器 veganFriendly 的值可以为 true 或者 false，如果为 true 就意味着你应该只包括 veganFriendlyi 为 true 的餐馆，为 false 则意味着可以包括任何餐馆。此外，我们还有最大价格 maxPrice 和最大距离 maxDistance 两个过滤器，它们分别考虑餐厅的价格因素和距离因素的最大值。
过滤后返回餐馆的 id，按照 rating 从高到低排序。如果 rating 相同，那么按 id 从高到低排序。简单起见， veganFriendlyi 和 veganFriendly 为 true 时取值为 1，为 false 时，取值为 0 。
示例 1：
输入：restaurants = [[1,4,1,40,10],[2,8,0,50,5],[3,8,1,30,4],[4,10,0,10,3],[5,1,1,15,1]], veganFriendly = 1, maxPrice = 50, maxDistance = 10
输出：[3,1,5]
解释：
这些餐馆为：
餐馆 1 [id=1, rating=4, veganFriendly=1, price=40, distance=10]
餐馆 2 [id=2, rating=8, veganFriendly=0, price=50, distance=5]
餐馆 3 [id=3, rating=8, veganFriendly=1, price=30, distance=4]
餐馆 4 [id=4, rating=10, veganFriendly=0, price=10, distance=3]
餐馆 5 [id=5, rating=1, veganFriendly=1, price=15, distance=1]
在按照 veganFriendly = 1, maxPrice = 50 和 maxDistance = 10 进行过滤后，我们得到了餐馆 3, 餐馆 1 和 餐馆 5（按评分从高到低排序）。
示例 2：
输入：restaurants = [[1,4,1,40,10],[2,8,0,50,5],[3,8,1,30,4],[4,10,0,10,3],[5,1,1,15,1]], veganFriendly = 0, maxPrice = 50, maxDistance = 10
输出：[4,3,2,1,5]
解释：餐馆与示例 1 相同，但在 veganFriendly = 0 的过滤条件下，应该考虑所有餐馆。
示例 3：
输入：restaurants = [[1,4,1,40,10],[2,8,0,50,5],[3,8,1,30,4],[4,10,0,10,3],[5,1,1,15,1]], veganFriendly = 0, maxPrice = 30, maxDistance = 3
输出：[4,5]
提示：
- 1 <= restaurants.length <= 104
- restaurants[i].length == 5
- 1 <= idi, ratingi, pricei, distancei <= 105
- 1 <= maxPrice, maxDistance <= 105
- veganFriendlyi 和 veganFriendly 的值为 0 或 1 。
- 所有 idi 各不相同。
方法：排序
按照题目要求，我们首先获取数组 restaurants 经过三个过滤器后的 filtered 数组。遍历 restaurants，假设当前遍历的元素为 restaurants[i]，如果以下条件都满足：
- restaurants[i][3] <= maxPrice
- restaurants[i][4] <= maxDistance
- veganFriendly 为 false 或 restaurants[i][2] 为 true
那么将 restaurants[i] 加入到 filtered 中。然后将 filtered 按照 rating 从高到低排序，如果 rating 相同，则按 id 从高到低排序。最后返回 filtered 的所有 id 作为结果。
代码：
class Solution {
public:
vector<int> filterRestaurants(vector<vector<int>>& restaurants, int veganFriendly, int maxPrice, int maxDistance) {
int n = restaurants.size();
vector<vector<int>> filtered;
for (int i = 0; i < n; i++) {
if (restaurants[i][3] <= maxPrice && restaurants[i][4] <= maxDistance && !(veganFriendly && !restaurants[i][2])) {
filtered.push_back(restaurants[i]);
}
}
sort(filtered.begin(), filtered.end(), [](vector<int> &v1, vector<int> &v2) -> bool {
return v1[1] > v2[1] || (v1[1] == v2[1] && v1[0] > v2[0]);
});
vector<int> res;
for (auto &v : filtered) {
res.push_back(v[0]);
}
return res;
}
};
时间 72ms 击败 86.81%使用 C++ 的用户
内存 27.50MB 击败 28.57%使用 C++ 的用户
复杂度分析
- 时间复杂度：O(nlogn)，其中 n 是数组 restaurants 的长度。排序需要 O(nlogn)。
- 空间复杂度：O(n)。保存 filtered 需要 O(n) 的空间。
author：力扣官方题解