Filter Restaurants by Vegan-Friendly, Price and Distance

class Solution {
public:
    vector<int> filterRestaurants(vector<vector<int>>& restaurants, int veganFriendly, int maxPrice, int maxDistance) {

        vector<vector<int>>ans;

        for(vector<int> v : restaurants){
            if((veganFriendly==0 || v[2]==veganFriendly) && v[3]<=maxPrice && v[4]<=maxDistance){
                ans.push_back(v);
            }
        }

        sort(ans.begin(),ans.end(),[](vector<int>a, vector<int>b){

            if(a[1]!=b[1])return a[1]>b[1];

            return a[0]>b[0];

        });

        vector<int> r;

        for(vector<int> a : ans)r.push_back(a[0]);

        return r;

    }
};