题目来源:烦恼的高考志愿 - 洛谷
这道题可以用二分查找解决。这题的目的是找到与学生的估分差距最小的学校并计算其总的满意度值,用二分查找可以降低一定查找速度。
主要步骤:
1、首先对学校的预计分线进行排序
2、用二分查找找到大于或者等于学生估分中差距最小的学校预计分数线mid
3、需要进行一定的特殊处理:
1、检查边界:如果循环结束时,可能存在越界的情况,需要判断如果 left 指向的是数组之外的元素,则选择数组中的最后一个元素。
2、如果left 是第一个元素,则直接用left相减。
3、否则,若left>1,需要考虑left-1和left之间那个比较小,选择最小的那个。
4、遍历完所有的学生分数,并计算累计的最小值。
#include<iostream>
#include<algorithm>
#include <cmath>
using namespace std;
#define maxn 1000010
int a[maxn], b[maxn], m, n;
int main()
{
cin >> m >> n;
for (int i = 1; i <= m; i++) {
cin >> a[i]; // 学校分数线
}
for (int j = 1; j <= n; j++) {
cin >> b[j]; // 学生估分
}
sort(a + 1, a + 1 + m);
long long ans = 0;
for (int i = 1; i <= n; i++)
{
int left = 1, right = m;
while (left <= right)
{
int mid = left + (right - left) / 2;
if (a[mid] <b[i])// 寻找大于等于 b[i] 的最小 a[mid]
{
left = mid + 1;
}
else
{
right = mid - 1;
}
}
// 如果 left 指向的是数组之外的元素,使用数组中的最后一个元素
if (left == m + 1) {
ans += b[i] - a[m];
}
else if (left == 1) { // 如果 left 是第一个元素
ans += a[left] - b[i];
}
else { // 否则,取最小的不满意度
ans += min(abs(a[left - 1] - b[i]), abs(a[left] - b[i]));
}
}
cout << ans << endl;
return 0;
}