题目来源:《计算机算法设计与分析》,王晓东
设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
第一次提交结果:
第一次提交的代码:
#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;
else if(x>a[n])cout<<n-1<<" "<<n;
else{
int low=0;
int mid=0;
int high =n-1;
while(low<=high){
mid=(low+high)/2;
if(a[mid]==x){
cout<<mid<<" "<<mid<<endl;
break;
}
if(x>a[mid]){
low=mid+1;
}
else{
high=mid-1;
}
}
if(low>high){
cout<<high<<" "<<low;
}
}
return 0;
}
第二次提交结果:
更正位置:i和j应该跟随mid更新,而不是直接等于low或high。
#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;
}