餐厅过滤器【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 )
空间复杂度:O ( log n )