二分查找算法之STL中lowerbound的妙用

题目背景

计算机竞赛小组的神牛V神终于结束了万恶的高考，然而作为班长的他还不能闲下来，班主任老t给了他一个艰巨的任务：帮同学找出最合理的大学填报方案。可是v神太忙了，身后还有一群小姑娘等着和他约会，于是他想到了同为计算机竞赛小组的你，请你帮他完成这个艰巨的任务。

题目描述

现有 m(m\le100000)m(m≤100000) 所学校，每所学校预计分数线是 a_i(a_i\le10^6)ai​(ai​≤106)。有 n(n\le100000)n(n≤100000) 位学生，估分分别为 b_i(b_i\le10^6)bi​(bi​≤106)。

根据n位学生的估分情况，分别给每位学生推荐一所学校，要求学校的预计分数线和学生的估分相差最小（可高可低，毕竟是估分嘛），这个最小值为不满意度。求所有学生不满意度和的最小值。

输入格式

第一行读入两个整数m,n。m表示学校数，n表示学生数。第二行共有m个数，表示m个学校的预计录取分数。第三行有n个数，表示n个学生的估分成绩。

输出格式

一行，为最小的不满度之和。

输入输出样例

输入 #1复制

4 3
513 598 567 689
500 600 550

输出 #1复制

32

说明/提示

数据范围：

对于30%的数据，m，n<=1000,估分和录取线<=10000;

对于100%的数据，n,m<=100,000,录取线<=1000000。

 

---

分析：

- 首先二话不说先排序，lower_boud和upper_bound都是在有序列表中才有用

- 本来以为要找到比估分低的学校，，，竟然要找离分数最近的数，不知道出题人是什么逻辑。。。离分数线近了就能上哪个？

- lower_bound的作用是在是个有序列表中，找到第一个大于等于要查找元素的迭代器，找不到返回该集合末尾的迭代器，数组中就是下标索引值。找到后只需与前一个元素比较，哪个接近分数选哪个学校

- 最后考虑两个边界(具体见代码)

    - 查找到的元素在第一个，即第一个就大于等于分数。

    - 没找到

#include <bits/stdc++.h>
using namespace std;

const int MAX = 1000005;

int M,N;
int a[MAX];

int main() {
    scanf("%d%d",&M,&N);
    for (int i = 0; i < M; ++i) {
        scanf("%d",&a[i]);
    }
    sort(a,a+M);
    int sum=0;
    for (int i = 0; i < N; ++i) {
        int x;
        scanf("%d",&x);
        int index = lower_bound(a,a+M,x)-a;
        if(index==M) {
            sum+=x-a[M-1];
            continue;
        }
        if(index==0) {
            sum+=a[0]-x;
            continue;
        }
        if(a[index]==x) continue;
        else if(a[index]>x){
            sum+=min(x-a[index-1],a[index]-x);
        }

    }
    printf("%d\n",sum);

}