餐厅过滤器【LC1333】
给你一个餐馆信息数组
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 。
-
思路
根据条件筛选餐厅,然后将餐厅按照规则排序,最后将id按照顺序放入结果集中
-
实现
class Solution { public List<Integer> filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) { List<int[]> list = new ArrayList<>(); for (int[] r : restaurants){ if (r[2] >= veganFriendly && r[3] <= maxPrice && r[4] <= maxDistance){ list.add(new int[]{r[0], r[1]}); } } list.sort((o1, o2) -> o1[1] == o2[1] ? o2[0] - o1[0] : o2[1] - o1[1]); List<Integer> res = new ArrayList<>(); for (int[] data : list){ res.add(data[0]); } return res; } }
- 复杂度
- 时间复杂度: O ( n ∗ log n ) \mathcal{O}(n*\log n) O(n∗logn)
- 空间复杂度: O ( log n ) \mathcal{O}(\log n) O(logn)
- 复杂度