【算法通关指南:算法基础篇】二分算法: 1.A-B 数对 2.烦恼的高考志愿

🔥小龙报:个人主页

🎬作者简介:C++研发,嵌入式,机器人等方向学习者

❄️个人专栏:《C语言》《【初阶】数据结构与算法》
永远相信美好的事情即将发生

文章目录


前言

本文将通过两道经典二分查找例题 ------A-B 数对与烦恼的高考志愿,带你系统掌握二分查找的核心思想与实用技巧。从排序预处理到lower_bound、upper_bound的灵活运用,再到手动实现二分与边界细节处理,由浅入深讲解算法原理与代码实现,帮助你快速攻克二分查找题型,提升编程思维与解题效率


一、A-B 数对

1.1题目

链接:A-B 数对

1.2 算法原理

由于顺序不影响最终结果,所以可以先把整个数组排序,来研究是佛否有其他的性质。

由A − B = C 得:B = A − C,由于C是已知的数,我们可以从前往后枚举所有的A ,然后去前面找有多少个符合要求的B ,正好可以用二分快速查找出区间的长度。

【STL使用】

1.lower_bound:传入要查询区间的左右迭代器(注意是左闭右开的区间,如果是数组就是左右指针)以及要查询的值k,然后返回该数组中 >= k的第一个位置

2.upper_bound :传入要查询区间的左右迭代器(注意是左闭右开的区间,如果是数组就是左右指针)以及要查询的值k ,然后返回该数组中 > k 的第⼀个位置;

例:a = 10, 20, 20, 20, 30, 40 ,设下标从1 开始计数,在整个数组中查询20 :

  1. lower_bound(a + 1, a + 1 + 6, 20) ,返回a + 2 位置的指针;
  2. upper_bound(a + 1, a + 1 + 6, 20) ,返回a + 5 位置的指针;
  3. 然后两个指针相减,就是包含20 这个数区间的长度。

1.3代码

coffeescript 复制代码
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 2e5 + 10;
LL a[N];

int main()
{
	int n, c;
	cin >> n >> c;
	for (int i = 1; i <= n; i++)
		cin >> a[i];

	sort(a + 1,a + 1 + n);

	LL ret = 0;
	for (int i = 2; i <= n; i++)
	{
		LL b = a[i] - c;
		ret += upper_bound(a + 1,a + i + 1,b) - lower_bound(a + 1, a + 1 + i, b);
	}

	cout << ret << endl;
	return 0;
}

注:同时STL的使用范围很「局限」,查询「有序序列」的时候才有用,数组无序的时候就无法使用。但是我们的二分算法也能在「数组无序」的时候使用,只要有「二段性」即可

二、烦恼的高考志愿

2.1 题目

链接:烦恼的高考志愿

2.2 算法原理

先把学校的录取分数「排序」,然后针对每一个学生成绩 ,在「录取分数」中二分出≥ b的「第一个」位置pos,那么差值最小的结果要么在pos位置,要么在位置pos - 1,那么最后的不满意度求法就为:

abs(apos − b)与abs(apos − 1 − b)的最小值。

注:细节问题

如果所有元素都大于b的时候,pos − 1 会在0 下标的位置,有可能结果出错;

• 如果所有元素都小于pos的时候,pos会在n的位置,此时结果倒不会出错,但是我们要想到这个细节问题,这道题不出错不代表下⼀道题不出错。

解决方法:加上两个左右护法,结果就不会出错了

2.3 代码

coffeescript 复制代码
//烦恼的高考志愿
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
LL a[N],b[N],m,n;

LL find(LL x)
{
	int l = 1, r = m;
	while (l < r)
	{
		LL mid = (l + r) / 2;
		if (a[mid] >= x)
			r = mid;
		else
			l = mid + 1;
	}
	return l;
}

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

	for (int i = 1; i <= m; i++)
		cin >> a[i];
	//设置左护法
	a[0] = -1e7;
	sort(a + 1, a + 1 + m);
	
	LL ret = 0;
	for (int i = 1; i <= n; i++)
	{
		cin >> b[i];
		LL pos = find(b[i]);
		ret += min(abs(a[pos - 1] - b[i]),abs(a[pos] - b[i]));
	}

	cout << ret << endl;
	return 0;
}

总结与每日励志

✨通过两道题目,我们学会了利用排序与二分查找快速统计数对、寻找最优匹配,理解了边界处理与 STL 函数的正确使用。二分的本质是寻找二段性,代码简洁却考验思维。愿你在算法学习路上保持耐心与专注,每一次思考都在沉淀成长,永远相信美好的事情即将发生,坚持下去,终会遇见更优秀的自己。

相关推荐
用户9385156350712 小时前
从 O(n²) 到 O(nlogn):一文读懂快速排序的“快”与“妙”
javascript·算法
To_OC13 小时前
手写快排次次翻车?别死背快排模板了,这才是面试官想听的底层逻辑
javascript·算法·排序算法
饼干哥哥14 小时前
Reddit VOC调研太慢?搭一个AI专家团队半小时洞察任何品类|以猫用饮水机为例
人工智能·算法·ai编程
地平线开发者15 小时前
Transformer模型部署之性能优化指南
算法
地平线开发者15 小时前
人在途中:从“编译失败”到“模型可落地”——CUDA 自定义算子
算法·自动驾驶
半个落月18 小时前
从递归到快速排序:用 JavaScript 把分治思想讲明白
javascript·算法·面试
小月土星19 小时前
JavaScript 快速排序:从 pivot、双指针到分治思想
javascript·算法·面试
小月土星19 小时前
JavaScript 递归入门:从 1 到 n 求和,再到数组扁平化
javascript·算法·面试
To_OC1 天前
LC 1 两数之和:面试第一道必考题,暴力解法直接被面试官 pass
javascript·算法·leetcode