python实现一可升降式的冒泡排序

前几天看了个冒泡的排序，这里手敲了一下代码，简单记录如下：

def bubble_sort(arr, order='asc'):
	# #可选升降序的冒泡排序, order>0升序，order<0降序
	# swapNeeded = True  # flag位
	counter = 0  # 计数(用于统计多少次排好序)
	for i in range(len(arr)):  # 这个循环负责设置冒泡排序进行的次数（比如说n个数，则只要进行n-1次冒泡，就可以把这个n个数排序好）

		swap_needed = False
		for j in range(0, len(arr)-i-1):
			# 这里这个j呢就是控制每一次具体的冒泡过程，请你想一想，我们第一次冒泡需要冒几次，也就是说需要比较几次，
			# 假如有三个数，那只需要两次就可以了，当下一次时，最后一个
			# 已经是有序的了，所以说少冒泡一次，所以这里j每次都会减去i的值，即不用冒“无用之泡泡”

			if order == 'asc':
				if arr[j] > arr[j+1]:
					arr[j], arr[j + 1] = arr[j + 1], arr[j]
					swap_needed = True

			elif order == 'desc':
				if arr[j] < arr[j+1]:
					arr[j], arr[j + 1] = arr[j + 1], arr[j]
					swap_needed = True

		if not swap_needed:
			break
		i += 1
		counter += 1

	return arr, counter


arr = [34,2,5,4,17,36,8]
# print(bubble_sort(arr))
print(bubble_sort(arr, 'asc'))
print(bubble_sort(arr, 'desc'))

结果：

([2, 4, 5, 8, 17, 34, 36], 3)
([36, 34, 17, 8, 5, 4, 2], 6)

参考："菜鸟教程"
"python排序算法之一：冒泡排序（及其优化）"