设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
int i, j;
void search(int a[], int left, int right, int x){
while(left <= right){
int mid = (left + right) / 2;
if(a[mid] == x){
i = j = mid;
return;
}
else if(a[mid] > x) right = mid - 1;
else left = mid + 1;
}
i = right;
j = left;
}
int main(){
int n, x;
scanf("%d%d", &n, &x);
int a[n];
for(int i = 0; i < n; i++)
scanf("%d", a + i);
if(x < a[0]) printf("-1 0");
else if(x > a[n-1]) printf("%d %d", n-1, n);
else{
search(a, 0, n - 1, x);
printf("%d %d", i, j);
}
return 0;
}
简化版
#include
int i, j;
void search(int a[], int left, int right, int x){
while(left <= right){
int mid = (left + right) / 2;
if(a[mid] == x){
i = j = mid;
return;
}
else if(a[mid] > x) right = mid - 1;
else left = mid + 1;
}
i = right;
j = left;
}
int main(){
int n, x;
scanf("%d%d", &n, &x);
int a[n];
for(int i = 0; i < n; i++)
scanf("%d", a + i);
search(a, 0, n - 1, x);
printf("%d %d", i, j);
return 0;
}