leetcode1333

class Solution:
    def filterRestaurants(self, restaurants: 'List[List[int]]', veganFriendly: int, maxPrice: int, maxDistance: int) -> 'List[int]':
        n = len(restaurants)
        satisfied = []
        for i in range(n):
            cur = restaurants[i]
            if veganFriendly == 1:
                if cur[2] == 1 and cur[3] <= maxPrice and cur[4] <= maxDistance:
                    satisfied.append([cur[0],cur[1]])
            else:
                if cur[3] <= maxPrice and cur[4] <= maxDistance:
                    satisfied.append([cur[0],cur[1]])
        satisfied = sorted(satisfied,key=lambda x:[-x[1],-x[0]])
        result = []
        for s in satisfied:
            result.append(s[0])
        return result

算法思路：多维数组，多条件排序。

先取出所有符合条件的元素，再按照两个条件排序。