设a[0:n-1]是已排好序的数组,请改写二分搜索算法,使得当x不在数组中时,返回小于x的最大元素位置i和大于x的最小元素位置j。当搜索元素在数组中时,i和j相同,均为x在数组中的位置。
输入格式:
输入有两行:
第一行是n值和x值;
第二行是n个不相同的整数组成的非降序序列,每个整数之间以空格分隔。
输出格式:
输出小于x的最大元素的最大下标i和大于x的最小元素的最小下标j。当搜索元素在数组中时,i和j相同。
提示:若x小于全部数值,则输出:-1 0
若x大于全部数值,则输出:n-1的值 n的值
输入样例:
在这里给出一组输入。例如:
6 5
2 4 6 8 10 12
输出样例:
在这里给出相应的输出。例如:
1 2
测试点:
1、查找第一个数
2、查找最后一个数
3、x不存在且小于全部数
4、x不存在且大于全部数
一、递归算法
#include <iostream>
using namespace std;
void Search(const int a[], int low, int high, int x, int& i, int& j) {
if (low > high) {
return; // 递归结束条件
}
int mid = (low + high) / 2;
if (a[mid] == x) {
i = j = mid; // 找到 x
} else if (a[mid] < x) {
i = mid; // 更新小于 x 的最大元素位置
Search(a, mid + 1, high, x, i, j);
} else {
j = mid; // 更新大于 x 的最小元素位置
Search(a, low, mid - 1, x, i, j);
}
}
int main() {
int n, x;
cin >> n >> x;
int a[n];
for (int i = 0; i < n; i++) {
cin >> a[i];
}
if (x < a[0]) {
cout << -1 << " " << 0 << endl; // 若 x 小于全部数值
} else if (x > a[n - 1]) {
cout << n - 1 << " " << n << endl; // 若 x 大于全部数值
} else {
int i = -1, j = n;
Search(a, 0, n - 1, x, i, j);
cout << i << " " << j << endl; // 输出小于 x 的最大位置和大于 x 的最小位置
}
return 0;
}
二、非递归算法
#include <iostream>
using namespace std;
int main() {
int n, x;
cin >> n >> x;
int a[n];
for (int i = 0; i < n; i++) {
cin >> a[i];
}
if (x < a[0]) {
cout << -1 << " " << 0 << endl; //若x小于全部数值,则输出:-1 0
} else if (x > a[n - 1]) {
cout << n - 1 << " " << n << endl; //若x大于全部数值,则输出:n-1的值 n的值
} else {
int low = 0, high = n - 1;
int i = -1, j = n;
while (low <= high) {
int mid = (low + high) / 2;
if (a[mid] == x) {
cout << mid << " " << mid << endl;
return 0; // 找到x,直接返回
}
if (a[mid] < x) {
i = mid; // 更新小于x的最大元素位置
low = mid + 1;
} else {
j = mid; // 更新大于x的最小元素位置
high = mid - 1;
}
}
cout << i << " " << j << endl; // 输出小于x的最大位置和大于x的最小位置
}
return 0;
}