1333. Filter Restaurants by Vegan-Friendly, Price and Distance

Given the array `restaurants` where  `restaurants[i] = [idi, ratingi, veganFriendlyi, pricei, distancei]`. You have to filter the restaurants using three filters.

The `veganFriendly` filter will be either true (meaning you should only include restaurants with `veganFriendlyi` set to true) or false (meaning you can include any restaurant). In addition, you have the filters `maxPrice` and `maxDistance` which are the maximum value for price and distance of restaurants you should consider respectively.

Return the array of restaurant IDs after filtering, ordered by rating from highest to lowest. For restaurants with the same rating, order them by id from highest to lowest. For simplicity `veganFriendlyi` and `veganFriendly` take value 1 when it is true, and 0 when it is false.

Example 1:

```Input: 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
Output: [3,1,5]
Explanation:
The restaurants are:
Restaurant 1 [id=1, rating=4, veganFriendly=1, price=40, distance=10]
Restaurant 2 [id=2, rating=8, veganFriendly=0, price=50, distance=5]
Restaurant 3 [id=3, rating=8, veganFriendly=1, price=30, distance=4]
Restaurant 4 [id=4, rating=10, veganFriendly=0, price=10, distance=3]
Restaurant 5 [id=5, rating=1, veganFriendly=1, price=15, distance=1]
After filter restaurants with veganFriendly = 1, maxPrice = 50 and maxDistance = 10 we have restaurant 3, restaurant 1 and restaurant 5 (ordered by rating from highest to lowest).
```

Example 2:

```Input: 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
Output: [4,3,2,1,5]
Explanation: The restaurants are the same as in example 1, but in this case the filter veganFriendly = 0, therefore all restaurants are considered.
```

Example 3:

```Input: 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
Output: [4,5]
```

Constraints:

• `1 <= restaurants.length <= 10^4`
• `restaurants[i].length == 5`
• `1 <= idi, ratingi, pricei, distancei <= 10^5`
• `1 <= maxPrice, maxDistance <= 10^5`
• `veganFriendlyi` and `veganFriendly` are 0 or 1.
• All `idi` are distinct.

1333. Filter Restaurants by Vegan-Friendly, Price and Distance
``````struct Solution;
use std::cmp::Reverse;

type Pair = (Reverse<i32>, Reverse<i32>);
impl Solution {
fn filter_restaurants(
restaurants: Vec<Vec<i32>>,
vegan_friendly: i32,
max_price: i32,
max_distance: i32,
) -> Vec<i32> {
let mut pairs: Vec<Pair> = restaurants
.into_iter()
.filter_map(|r| {
if (vegan_friendly == 0 || r[2] == 1) && r[3] <= max_price && r[4] <= max_distance {
Some((Reverse(r[1]), Reverse(r[0])))
} else {
None
}
})
.collect();
pairs.sort_unstable();
pairs.into_iter().map(|p| (p.1).0).collect()
}
}

#[test]
fn test() {
let restaurants = vec_vec_i32![
[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]
];
let vegan_friendly = 1;
let max_price = 50;
let max_distance = 10;
let res = vec![3, 1, 5];
assert_eq!(
Solution::filter_restaurants(restaurants, vegan_friendly, max_price, max_distance),
res
);
let restaurants = vec_vec_i32![
[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]
];
let vegan_friendly = 0;
let max_price = 50;
let max_distance = 10;
let res = vec![4, 3, 2, 1, 5];
assert_eq!(
Solution::filter_restaurants(restaurants, vegan_friendly, max_price, max_distance),
res
);
let restaurants = vec_vec_i32![
[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]
];
let vegan_friendly = 0;
let max_price = 30;
let max_distance = 3;
let res = vec![4, 5];
assert_eq!(
Solution::filter_restaurants(restaurants, vegan_friendly, max_price, max_distance),
res
);
}
``````