改写二分搜索算法

描述

设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

solution

#include <stdio.h>
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 <stdio.h>
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;
}