c++二维vector用特定元素排序时sort中cmp的重写

最新推荐文章于 2025-09-10 10:29:17 发布

原创最新推荐文章于 2025-09-10 10:29:17 发布 · 1.5k 阅读

2 ·

CC 4.0 BY-SA版权

C++ 同时被 2 个专栏收录

13 篇文章

订阅专栏

LeetCode

13 篇文章

订阅专栏

本文介绍了一种使用C++实现的餐厅过滤算法，通过两个方法对比，详细解释了如何根据特定条件筛选餐厅，并展示了方法1相较于方法2在运行效率上的优势。关键在于避免不必要的函数调用，从而减少运行时间。

在这里插入图片描述

Method 1：

class Solution {
    public:
    vector<int> filterRestaurants(vector<vector<int>>& restaurants, int veganFriendly, int maxPrice, int maxDistance) {
        int size=restaurants.size();
        vector<vector<int>>res;
        for(int i=0;i<size;i++)
        {
            if((veganFriendly==1&&restaurants[i][2]==0)||restaurants[i][3]>maxPrice||restaurants[i][4]>maxDistance)
            {
                continue;
            }
            res.push_back(restaurants[i]);
        }
        sort(res.begin(),res.end(),[](const auto& a,const auto& b){
            return (a[1]>b[1])||(a[1]==b[1]&&a[0]>b[0]);
        });
        size=res.size();
        vector<int>ans;
        for(int i=0;i<size;i++){
                ans.push_back(res[i][0]);
        }
        return ans;
    }
};

Method 2：

class Solution {
    public:
    static bool cmp(const vector<int>&a,const vector<int>&b){
        return a[1]>b[1]||(a[1]==b[1]&&a[0]>b[0]);
    }
    vector<int> filterRestaurants(vector<vector<int>>& restaurants, int veganFriendly, int maxPrice, int maxDistance) {
        int size=restaurants.size();
        vector<vector<int>>res;
        for(int i=0;i<size;i++)
        {
            if((veganFriendly==1&&restaurants[i][2]==0)||restaurants[i][3]>maxPrice||restaurants[i][4]>maxDistance)
            {
                continue;
            }
            res.push_back(restaurants[i]);
        }
        sort(res.begin(),res.end(),cmp);
        size=res.size();
        vector<int>ans;
        for(int i=0;i<size;i++){
                ans.push_back(res[i][0]);
        }
        return ans;
    }
};