1. 题目说明
2. 题目解析
二分是算法中很重要的思想,主要分为两大类:整数二分、浮点数二分。
其中,浮点数二分比较简单,多见于求平方根、三次方根等题目。整数二分最复杂,各种边界情况,以及到底是大于,还是大于等于,到底是更新左端点,还是更新右端点,到底是 l = mid
,还是 l = mid + 1
。
这些各式各类的边界问题,让写整数二分题目成为了一种奇妙的体验:我也不知道怎么了,反正代码 AC
了。玄学 AC
…
在此,整理下整数二分的模板题,这道经典问题可以覆盖所有的整数二分问题了。
模板代码如下:
// 涵盖绝绝绝大多数整数二分问题的一个成熟模板
bool check(int x) {/* ... */} // 检查x是否满足某种性质
// 区间[l, r]被划分成[l, mid]和[mid + 1, r]时使用:
int bsearch_1(int l, int r)
{
while (l < r)
{
int mid = l + r >> 1;
if (check(mid)) r = mid; // check()判断mid是否满足性质
else l = mid + 1;
}
return l;
}
// 区间[l, r]被划分成[l, mid - 1]和[mid, r]时使用:
int bsearch_2(int l, int r)
{
while (l < r)
{
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
return l;
}
问题代码如下:
#include <iostream>
using namespace std;
const int N = 1e5+5;
int n, m;
int q[N];
int main() {
cin >> n >> m;
for (int i = 0; i < n; ++i) cin >> q[i];
while (m --) {
int x;
cin >> x;
int l = 0, r = n - 1;
while (l < r) {
int mid = l + r >> 1;
if (q[mid] >= x) r = mid;
else l = mid + 1;
}
// 在此没找到的话是会找到了第一个比x大的数字,可复用
if (q[l] != x) cout << -1 << ' ' << -1 << endl;
else {
cout << l << ' ';
int l = 0, r = n - 1;
while (l < r) {
int mid = l + r + 1 >> 1;
if (q[mid] <= x) l = mid;
else r = mid - 1;
}
cout << r << endl;
}
}
return 0;
}