算法的原理和步骤:
二分查找是一种基于比较目标值和数组中间元素的值来决定搜索范围的算法。如果当前元素等于目标值,则搜索结束;如果当前元素小于目标值,则在右侧继续搜索;如果当前元素大于目标值,则在左侧继续搜索。
下面是算法的步骤:
- 对于每个查询元素 k,进行二分查找:
- 初始化左边界 l 为 0,右边界 r 为 n-1,其中 n 是数组的长度。
- 在每一步中,计算中间位置 mid = (l + r) / 2。
- 比较目标元素 k 与数组中间位置的元素 q[mid]:
- 如果 k 等于 q[mid],则找到了目标元素,记录当前位置为起始位置,然后继续查找右侧的终止位置。
- 如果 k 小于 q[mid],则更新右边界 r 为 mid - 1,向左侧继续查找。
- 如果 k 大于 q[mid],则更新左边界 l 为 mid + 1,向右侧继续查找。
- 在找到起始位置后,重置左边界 l 为 0,右边界 r 为 n-1,继续进行二分查找,寻找终止位置。
- 最终返回起始位置和终止位置。
这样的方法能够在 O(log n) 的时间内完成查询操作,因为二分查找每次可以将搜索范围减半。
问题:
给定升序排列的整数数组、数组长度n(1<= n <= 100000)和查询次数m(1<= m <= 10000),对于每个查询元素 x(1<= x <= 10000),返回其在数组中的起始和终止位置。如果数组中不存在该元素,则返回 -1 -1。
输入样例:
6 3
1 2 2 3 3 4
3
4
5
输出样例:
3 4
5 5
-1 -1
c++代码详情:
#include<iostream>
using namespace std;
const int N = 100010;
int n, m, q[N]; // 定义数组大小为 N,n 表示数组 q 的长度,m 表示查询次数
int main() {
cin >> n >> m; // 输入数组长度和查询次数
for(int i = 0; i < n; i++) cin >> q[i]; // 输入数组元素
// 处理查询
while(m--) {
int x;
cin >> x; // 输入查询的值 x
// 二分查找左边界
int l = 0, r = n - 1;
while(l < r) {
int mid = l + r >> 1; // 取中间值
if(q[mid] >= x) r = mid; // 如果中间值大于等于 x,更新右边界
else l = mid + 1; // 否则更新左边界
}
// 如果找不到 x,则输出 "-1 -1"
if(q[l] != x) puts("-1 -1");
else {
cout << l << " "; // 输出左边界的位置
// 重新初始化左右边界,进行二分查找右边界
l = 0, r = n - 1;
while(l < r) {
int mid = l + r + 1 >> 1; // 注意取右中位数
if(q[mid] <= x) l = mid; // 如果中间值小于等于 x,更新左边界
else r = mid - 1; // 否则更新右边界
}
cout << l << endl; // 输出右边界的位置
}
}
return 0;
}
java代码详情:
import java.util.Scanner;
public class Main{
final static int N = 100010;
static int[] binary = new int[N];
public static void binaryQuery(int x,int n){
int l = 0,r = n - 1;
while(l < r)
{
int mid = l + r >> 1;
if(binary[mid] >= x) r = mid;
else l = mid + 1;
}
if(binary[l] != x) System.out.println("-1 -1");
else{
System.out.printf("%d ",l);
l = 0;
r = n - 1;
while(l < r)
{
int mid = l + r + 1 >> 1;
if(binary[mid] <= x) l = mid;
else r = mid - 1;
}
System.out.printf("%d\n",l);
}
}
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int q = sc.nextInt();
for(int i = 0;i < n;i ++) binary[i] = sc.nextInt();
while(q -- != 0)
{
int x = sc.nextInt();
binaryQuery(x,n);
}
}
}