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

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

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

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

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

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


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

#include<iostream>
#include<vector>
#include<algorithm>
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<heaps> 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;
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 前言 把《C++ Primer》[https://book.douban.com/subject/25708312...
    尤汐Yogy阅读 9,533评论 1 51
  • 〇、前言 本文共108张图,流量党请慎重! 历时1个半月,我把自己学习Python基础知识的框架详细梳理了一遍。 ...
    Raxxie阅读 19,024评论 17 410
  • 前段时间网络上报出一医院为抢救重病患者争分夺秒,把患者衣服剪了,之后患者家属要求索赔的情况,虽然大家都力挺医生,为...
    十三号仓库阅读 336评论 4 6
  • dns查询过程解析 DNS查询模式有递归查询和迭代查询,他们的区别如下: (1)递归查询 递归查询是一种DNS 服...
    think_lonely阅读 836评论 0 1
  • 在绝对的实力差异面前,技巧只是自娱自乐和自我消耗罢了。 下面还是结合思维导图,分成基本思想、科学水平与技术、面壁计...
    陶薰读书阅读 2,529评论 2 4