二分是一种在实现上非常简单的算法,但它主要难在思想上,在分类讨论处理边界问题的时候,有清晰明了的思路,会让代码完成效率大大提升,建议处理时用最好辅以画图来帮助思考。
接下来时几道简单的二分题,用来体会不同情况时处理的思想。
Q1:
原题链接:https://nanti.jisuanke.com/t/T1560
蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问整数 x 是否在数组 A 中。
输入格式
第一行输入两个整数 n 和 m,分别表示数组的长度和查询的次数。
接下来一行有 n 个整数 ai。
接下来 m 行,每行有 1 个整数 x,表示蒜头君询问的整数。
输出格式
对于每次查询,如果可以找到,输出"YES"
,否则输出"NO"
。
数据范围
1 ≤ n, m ≤ 10^5, 0 ≤ x ≤ 10^6。
样例
Sample 1
Inputcopy | Outputcopy |
---|---|
10 5 1 1 1 2 3 5 5 7 8 9 0 1 4 9 10 | NO YES NO YES NO |
注意: 题目之中并没有明确指出所给数组为有序数组,所以要对数组先进行排序操作。
核心代码:
int l = 0,r = n - 1;
//先声明两边界
while (l < r)
//当不能再分时,即 l = r 时退出循环
{
int mid = l + r >> 1;
//声明的中间值,随着l与r的更新而更新
if(q[mid] >= x) r = mid;
else l = mid + 1;
//最核心部分,要想清楚符合条件后如何更新边界,不符合条件的话更新边界要注意什么
}
当满足条件的时候,x 介于 l 与 mid 之间,由于存在 q [ mid ] = x 的可能,所有更新边界 r 的时候,新的边界要包含 mid ; 相反,由于 q [ mid ] < x,不存在 x 取到 q [ mid ] 的可能,所以更新边界 l 时候便从 mid 的下一位开始。
只要想清楚如何处理边界,剩下便的十分简单了。
完整代码如下:
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10;
int m, n;
int q[N];
int main()
{
while(~scanf("%d %d",&n, &m) != 0)
{
for(int i = 0; i < n; i++) scanf("%d",&q[i]);
sort(q, q + n);
for(int i = 0; i < m; i++)
{
int x;
scanf("%d", &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;
}
if(q[l] == x) printf("YES\n");
else printf("NO\n");
}
}
return 0;
}