自定义lower_bound二分查找函数的比较方式(网易2019秋招笔试题1)

第一种方式,为参与比较的元素类型重载<

第二种方式,最后增加一个仿函数临时对象参数

第三种方式,最后增加一个函数指针参数

特别需要注意的就是,传入的要查找的参数类型也一定是和容器中的数据类型一样,而不是和具体比较的时候用的基本类型一样,比如下面的例子,不能直接将int型的qi传入查找,而是应该构建一个临时对象heaps(0,qi),再去和其他的容器中元素进行比较。

/*
问题:
将n堆苹果,从左往右摆放,并对苹果和堆都从1开始顺序编号。
现在任意喊一个苹果编号,问你是在第几堆。
输入:一个堆数n,然后一行是n个数表示每堆苹果的数量。
再输入一个数m表示有几个苹果要找,然后是m个苹果编号。
输出:m个苹果编号对应的是在第几堆。


分析:就是一个二分法的典型应用。首先要对堆用一个id记录下来,然后要在堆中记录这个堆表示的边界值。
将各个苹果堆插入到一个vector中
然后根据堆类中的边界值对vector排序,然后再对有序序列用lower_bound二分查找函数进行查找,
返回的就是大于等于要查找值的边界值,对应输出该堆的id,就是我们要的结果。
*/

#include
#include
#include
using namespace std;
class heaps
{
public:
    int id;
    int num;
    heaps(int i, int j) :id(i), num(j){};
    //方式一:重载小于号的形式,注意只能有一个参数,<号分左值和右值
    
    bool operator<(const heaps &h2)
    {
        return num < h2.num;
    }
    

};
//仿函数形式的比较函数,使用的时候传入临时对象comp()
class comp
{
public:
    bool operator()(const heaps &a, const heaps &b)
    {
        return a.num < b.num;
    }
};


//纯比较函数,使用的时候传入函数名就行
bool comp_func(const heaps &a, const heaps &b)
{
return a.num < b.num;
}


int main()
{
    int n;
    cin >> n;

    vector v_apps;
    int a1;
    cin >> a1;
    v_apps.push_back(heaps(1, a1));

    for (int i = 1; i < n; i++)
    {
        int ai;
        cin >> ai;
        int aa = ai + v_apps[i - 1].num;
        v_apps.push_back(heaps(i + 1, aa));
    }

    int m;
    cin >> m;
    for (int i = 0; i < m; i++)
    {
        int qi;
        cin >> qi;
        //第一种用法,特别注意,不能直接用qi作为参数,而是应该构造一个heaps类的临时对象,才能比较
        auto pos = lower_bound(v_apps.begin(), v_apps.end(), heaps(0, qi));
        //第二种用法
        //auto pos = lower_bound(v_apps.begin(), v_apps.end(), heaps(0, qi), comp());
        //第三种用法
        //auto pos = lower_bound(v_apps.begin(), v_apps.end(), heaps(0, qi), comp_func);
        cout << (*pos).id << endl;
    }

    return 0;
}

你可能感兴趣的:(自定义lower_bound二分查找函数的比较方式(网易2019秋招笔试题1))