二分问题的要点:
首先在二分查找前,数组肯定是升序排序的。假设我们想要寻找的数是3并且数组中存在多个3,我们可以分别使用两个模版找到最左边的3和最右边的3。
1. 判断使用哪个模板:
(1)便捷记法:
- 查找第一个大于等于5的数的下标 if (a[mid] >= 5) r = mid;
- 查找最后一个小于等于5的数的下标 if (a[mid] <= 5) l = mid;
(2)比较繁琐(详细)的记法:
- 如果要找左区间的右端点,用模板一;要找右区间的左端点,用模板二。
- 对于这道题来说,首先要确定数x的左边界,可以知道这个数的右边满足所有的数都大于等于x的性质,而左边都不满足这个性质,符合图中绿色,找左边界用模板二。然后确定数x的右边界,可以知道这个数左边的数都满足小于等于x的性质,而右边都都不满足这个性质,符合图中红色,找右边界用模板一。
2. 区间更新:
- l = mid,mid = l + r + 1 >> 1;
- r = mid, mid = l + r >> 1;
3. 二分查找完毕后l == r,同时指向答案:
- 若在数组中找到该数,返回 l 或 r 就是返回该数的下标。
- 若该数不存在,那么 l 指向的数不等于想要寻找的数。
代码:
#include <iostream>
using namespace std;
const int N = 1e5 + 10;
int q[N];
int main()
{
int n, m;
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;
// 找第一个大于等于3的点的下标,也就是3的左区间
while (l < r)
{
int mid = l + r >> 1;
// 要找最左边的3,因为左边3的右边都满足大于等于3的性质,而它的左边都不满足这个性质
// 所以假设mid满足这个性质. 若mid满足这个性质,要找到的3在mid左边,所以r=mid
if (q[mid] >= x) r = mid;
else l = mid + 1;
}
// 二分查找结束时l等于r,且q[l]或q[r]就是要找的那个数,l或r代表该数的下标
if (q[l] != x) cout << "-1 -1" << endl;
else
{
cout << l << " ";
int l = 0, r = n - 1;
// 找最后一个小于等于3的点的下标,也就是3的右区间
while (l < r)
{
// 要找最右边的3,因为右边3的左边都满足小于等于3的性质,而它的右半边都不满足这个性质
// 所以假设mid满足小于等于3的性质.若mid满足这个性质,要找到的3在mid右边,所以l=mid
int mid = l + r + 1 >> 1;
if (q[mid] <= x) l = mid;
else r = mid - 1;
}
cout << l << endl;
}
}
return 0;
}