【排序-中等】1333. 餐厅过滤器

最新推荐文章于 2025-10-20 14:14:17 发布

原创最新推荐文章于 2025-10-20 14:14:17 发布 · 138 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

714 篇文章

订阅专栏

本文介绍了一种用于筛选餐馆信息的算法，该算法可以根据特定条件（如是否为素食友好、最大价格和最大距离）过滤并排序餐馆。通过示例展示了不同参数组合下筛选结果的变化。

【题目】
给你一个餐馆信息数组 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 <= 10^4
restaurants[i].length == 5
1 <= idi, ratingi, pricei, distancei <= 10^5
1 <= maxPrice, maxDistance <= 10^5
veganFriendlyi 和 veganFriendly 的值为 0 或 1 。
所有 idi 各不相同。
【代码】
【Python】
执行用时：
124 ms, 在所有 Python3 提交中击败了7.69%的用户
内存消耗：
19.3 MB, 在所有 Python3 提交中击败了70.51%的用户

class Solution:
    def filterRestaurants(self, restaurants: List[List[int]], veganFriendly: int, maxPrice: int, maxDistance: int) -> List[int]:
        restaurants.sort(key=lambda x:(x[1],x[0]),reverse=True)
        i=0
        while i<len(restaurants):
            if (veganFriendly==1 and restaurants[i][2]!=veganFriendly) or restaurants[i][3]>maxPrice or restaurants[i][4]>maxDistance:
                print(restaurants[i])
                restaurants.pop(i);
                continue
            i+=1
        return [x[0] for x in restaurants]

【filter】
构造3个filter对list进行过滤，语法为filter(function, iterable)，用list()方法包裹即可返回原列表过滤结果

class Solution:
    def filterRestaurants(self, restaurants: List[List[int]], veganFriendly: int, maxPrice: int, maxDistance: int) -> List[int]:
        def is_veganFriendly(rest):
            if veganFriendly == 1:
                return rest[2] == 1
            else:
                return 1

        def is_less_maxPrice(rest):
            return rest[3] <= maxPrice

        def is_less_maxDistance(rest):
            return rest[4] <= maxDistance

        restaurants = list(filter(is_veganFriendly, restaurants))
        restaurants = list(filter(is_less_maxPrice, restaurants))
        restaurants = list(filter(is_less_maxDistance, restaurants))

        return [i[0] for i in sorted(restaurants, key=lambda x: (x[1],x[0]), reverse=True)]