题目:
Solution:
#include <iostream>
using namespace std;
const int N = 1e5 + 10;
int n, q;
int a[N];
int find(int k, int type) //type为0表示查找起始位置,type为1表示查找终止位置
{
int l = 0, r = n - 1, ans = -1;
while (l <= r)
{
int mid = l + r >> 1;
if (a[mid] >= k)
{
if (a[mid] == k)
ans = mid;
r = mid - 1;
}
else
l = mid + 1;
}
if (type == 0)
return ans;
l = 0, r = n - 1, ans = -1;
while (l <= r)
{
int mid = l + r >> 1;
if (a[mid] <= k)
{
if (a[mid] == k)
ans = mid;
l = mid + 1;
}
else
r = mid - 1;
}
return ans;
}
int main()
{
cin >> n >> q;
for (int i = 0; i < n; i ++ )
cin >> a[i];
while (q -- )
{
int k;
cin >> k;
int l = find(k, 0), r = find(k, 1);
cout << l << ' ' << r << endl;
}
return 0;
}
二分法查找法。
二分法是一种高效的查找算法,适用于有序数组。
在该代码中,find
函数使用了两次二分查找。第一次查找是为了找到元素k的起始位置,通过不断更新左右边界来逼近目标位置。如果当前中间位置的元素大于等于k,则将右边界更新为mid-1,否则将左边界更新为mid+1。当找到元素k时,将结果记录在ans变量中。
第二次查找是为了找到元素k的终止位置。同样通过不断更新左右边界来逼近目标位置。如果当前中间位置的元素小于等于k,则将左边界更新为mid+1,否则将右边界更新为mid-1。当找到元素k时,将结果记录在ans变量中。
最后,将起始位置和终止位置分别作为函数的返回值,输出到标准输出流中。
这样就可以实现在升序排列的数组中使用二分法查找指定元素的起始位置和终止位置。
二分查找的时间复杂度为O(logn),其中n为数组的长度。通过二分法,可以快速找到目标元素在数组中的位置。