题目来源:《计算机算法设计与分析》,王晓东
设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
代码:
n, x = map(int, input().split(' '))
list_nums = input().split(' ')
a = -1
b = -1
for i in range(0, len(list_nums)):
list_nums[i] = int(list_nums[i])
if x < list_nums[0]:
print('-1 0')
elif x > list_nums[n - 1]:
print('{} {}'.format(n - 1, n))
else:
if x in list_nums:
a = list_nums.index(x)
b = list_nums.index(x)
else:
for i in range(0, len(list_nums)):
if list_nums[i] > x:
a = i-1
b = i
break
print('{} {}'.format(a, b))