python二分查找数列

1. python二分查找数列

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import random
import datetime

#生成随机数列
def randList (min, max, l):
min,max = (int(min),int(max)) if min <= max else (int(max),int(min))
l = int(abs(l)) if l else 0
list = []
for i in range(l):
list.append(random.randint(min,max))
return list

# 返回 x 在 arr 中的索引,如果不存在返回 -1
def binarySearch (arr, l, r, x):

# 基本判断
if r >= l:

mid = int(l + (r - l)/2)

# 元素正好在中间位置
if arr[mid] == x:
return mid

# 元素小于中间位置的元素,只需要再比较左边的元素
elif arr[mid] > x:
return binarySearch(arr, l, mid-1, x)

# 元素大于中间位置的元素,只需要再比较右边的元素
else:
return binarySearch(arr, mid+1, r, x)

else:
# 不存在
return -1



# 测试数组
arr = randList (1, 50, 20) # 生成随机数列 randList(最小数,最大数,数列长度)
arr.sort() # 排序
print ("随机数列为",arr)
x = 20

startTime = datetime.datetime.now() # 开始执行时间
# 调用二分查找
result = binarySearch(arr, 0, len(arr)-1, x)

if result != -1:
print ("元素 %d 在数组中的索引为 %d" %(x, result))
else:
print ("元素不在数组中")

endTime = datetime.datetime.now() # 结束时间
print ("执行时间%dms" % ((endTime - startTime).seconds * 1000 + (endTime - startTime).microseconds/1000))

欢迎关注我的其它发布渠道