第一种方式,为参与比较的元素类型重载<
第二种方式,最后增加一个仿函数临时对象参数
第三种方式,最后增加一个函数指针参数
特别需要注意的就是,传入的要查找的参数类型也一定是和容器中的数据类型一样,而不是和具体比较的时候用的基本类型一样,比如下面的例子,不能直接将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;
}