问:
假如有 10 个小盆友从左到右站成一排,个头不等。
老师想让他们按照个头从低到高站好。
从头到尾找一个个头最小的小盆友,然后把它和第一个小盆友交换。
然后从第二个小盆友开始采取同样的策略,这样一圈下来小盆友就有序了。
答:
选择排序
import random
def select_sort(seq):
n = len(seq)
for i in range(n-1):
# 我们假设当前下标的元素是最小的
min_idx = i
for j in range(i+1, n):
# 从 i 的后边开始找到最小的元素,得到它的下标
if seq[j] < seq[min_idx]:
# 一个 j 循环下来之后就找到了最小的元素它的下标
min_idx = j
if min_idx != i:
# swap
seq[i], seq[min_idx] = seq[min_idx], seq[i]
print(f"第{i+1}轮:", seq)
print("结 果:", seq)
seq = list(range(10))
random.shuffle(seq)
print("原 始:", seq)
select_sort(seq)
冒泡排序
import random
def bubble_sort(seq): # O(n^2), n(n-1)/2 = 1/2(n^2 + n)
n = len(seq)
for i in range(n-1):
for j in range(n-1-i):
# 这里之所以 n-1 还需要 减去 i 是因为每一轮冒泡最大的元素都会冒泡到最后,无需再比较
if seq[j] > seq[j+1]:
seq[j], seq[j+1] = seq[j+1], seq[j]
# 我打印出来让你看清楚每一轮最高、次高、次次高...的小朋友会冒泡到右边
print(f"第{i+1}轮:", seq)
print("结 果:", seq)
# 注意 python3 返回迭代器,所以我都用 list 强转了,python2 range 返回的就是 list
seq = list(range(10))
# shuffle inplace 操作,打乱数组
random.shuffle(seq)
print("原 始:", seq)
bubble_sort(seq)
问:
假如一个序列已经有序了呢,我们还需要从头找到尾吗?
答:
当然不用,折半(二分)是一种经典思想,即二分查找
。
def binary_search(sorted_array, val):
if not sorted_array:
return -1
beg = 0
end = len(sorted_array) - 1
while beg <= end:
mid = int((beg + end) / 2)
if sorted_array[mid] == val:
return mid
elif sorted_array[mid] > val:
end = mid - 1
else:
beg = mid + 1
return -1
a = list(range(10))
# 正常值
print(binary_search(a, 1))
print(binary_search(a, -1))
# 边界值
print(binary_search(a, 0))
问:什么是线性查找?
答:线性查找就是从头找到尾,直到符合条件了就返回。比如在一个 list 中找到一个等于 5 的元素并返回下标:
number_list = [0, 1, 2, 3, 4, 5, 6, 7]
def linear_search(value, iterable):
for index, val in enumerate(iterable):
if val == value:
return index
return -1
print(linear_search(5, number_list))