这道题是上篇文章题目的改动版,需要重新考虑边界问题,并且还需对特殊情况进行判断。
上篇文章链接:二分查找的简单例题(一)_m0_64194841的博客-CSDN博客
原题链接:https://nanti.jisuanke.com/t/T156
蒜头君手上有个长度为 n 的数组 A。由于数组实在太大了,所以蒜头君也不知道数组里面有什么数字,所以蒜头君会经常询问在数组 A 中,大于等于 x 的最小值是多大?
输入格式
第一行输入两个整数 n 和 m,分别表示数组的长度和查询的次数。
接下来一行有 n 个整数 ai。
接下来 m 行,每行有 1 个整数 x,表示蒜头君询问的整数。
输出格式
对于每次查询,如果可以找到,输出这个整数。
否则输出 −1。
数据范围
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 | 1 1 5 9 -1 |
核心代码:
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("%d\n",x);
else
{
if(q[l] >= x)printf("%d\n",q[l]);
else printf("-1\n");
}
完整代码如下:
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10;
int m;
int n;
int q[N];
int main()
{
int n,m;
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("%d\n",x);
else
{
if(q[l] >= x)printf("%d\n",q[l]);
else printf("-1\n");
}
}
}
return 0;
}