首页 > 技术文章 > Python 常用算法记录

cleven 2019-04-25 13:55 原文

一、递归

汉诺塔算法:把A柱的盘子,移动到C柱上,最少需要移动几次,大盘子只能在小盘子下面

1、当盘子的个数为n时,移动的次数应等于2^n – 1

2、描述盘子从A到C:

(1)如果A只有一个圆盘,可以直接移动到C; 
(2)如果A有N个圆盘,可以看成A有1个圆盘(底盘) + (N-1)个圆盘,首先需要把 (N-1) 个圆盘移动到 B,然后,将 A的最后一个圆盘移动到C,再将B的(N-1)个圆盘移动到C。

3、代码

 1 i = 0
 2 
 3 def move(n, a, b, c):
 4     global i
 5     if n == 1:
 6         i += 1
 7         print('移动第', i, '', a, '-->', c)
 8     else:
 9         # 1.把A柱上n-1个圆盘移动到B柱上
10         move(n - 1, a, c, b)
11         # 2.把A柱上最大的移动到C柱子上
12         move(1, a, b, c)
13         # 3.把B柱子上n-1个圆盘移动到C柱子上
14         move(n - 1, b, a, c)
15 
16 move(4, 'A', 'B', 'C')

 

 

 

二、常用排序算法

 

1、冒泡排序  Bubble Sort

(1)两两比较,大元素往后排,类似水里的气泡从下面浮上来

(2)代码

 1 def bubble_sort(alist):
 2     """冒泡排序"""
 3     n = len(alist)
 4     for j in range(n - 1):
 5         count = 0
 6         for i in range(0, n - 1 - j):
 7             if alist[i] > alist[i + 1]:
 8                 alist[i], alist[i + 1] = alist[i + 1], alist[i]
 9                 count += 1
10         # count为0,表示数据本来就是有序的,无需多比较
11         if 0 == count:
12             return
13 
14 
15 if __name__ == "__main__":
16     li = [8, 1, 5, 4, 3, 9, 0, 2, 7, 6]
17     bubble_sort(li)
18     print(li)

 

 

2、选择排序 Selection Sort

(1)设置一个标记位记录每一轮遍历后(内循环)最小数值的角标,在内循环一轮结束后,直接将标记位指向的元素,放到本轮遍历(内循环)的第一个位置。

(2)代码

 1 def select_sort(alist):
 2     """选择排序"""
 3     n = len(alist)
 4     for j in range(n-1):
 5         min_index = j # 这是标记位
 6         for i in range(j+1, n):
 7             if alist[min_index] > alist[i]:
 8                 min_index = i # 记录下最小的数值对应的角标
 9         # 本轮内循环结束,本轮比较的数组的第一个元素为 alist[j]
10         alist[j], alist[min_index] = alist[min_index], alist[j]
11 
12 
13 if __name__ == "__main__":
14     li = [54, 26, 93, 17, 77, 31, 44, 55, 20]
15     select_sort(li)
16     print(li)

 

 

 

 3、插入排序 Insertion Sort

(1)逐个取出元素(比如从第二个开始),和前面的每个元素比较,插入到合适的位置。这样,保证一轮比较后,取出元素位置的左边都是排序过的有序数据。

(2)代码

 1 def insert_sort(alist):
 2     """插入排序"""
 3     n = len(alist)
 4     # 需要取出每一个元素和左边的逐个比较,j就是每个元素的角标,控制外循环
 5     for j in range(1, n):
 6         # j = [1, 2, 3, n-1]
 7         # i 代表内层循环起始值,每一次外循环更新内循环的起始值
 8         i = j
 9         while i > 0 and alist[i] < alist[i-1]:
10             # 插入到前面的正确位置中,依然需要逐个比较、交换
11             alist[i], alist[i-1] = alist[i-1], alist[i]
12             i -= 1
13 
14 
15 if __name__ == "__main__":
16     li = [54, 26, 93, 17, 77, 31, 44, 55, 20]
17     insert_sort(li)
18     print(li)

 

 

 

 4、希尔排序 SHELL

(1)选择一个步长(gap) ,然后按间隔gap取元素,放到一个组里,以此将这个数据分成多组。每个组采用插入排序处理数据。紧接着步长逐渐变小,循环前面操作,直至为1。

(2)简单图示:

        

 (3)代码

 1 def shell_sort(alist):
 2     """希尔排序"""
 3     n = len(alist)
 4     gap = n // 2
 5     while gap > 0:
 6         # 插入算法,与普通的插入算法的区别就是gap步长
 7         for j in range(gap, n):
 8             # j = [gap, gap+1, gap+2, gap+3, ..., n-1]
 9             i = j
10 
11             while i > 0 and alist[i] < alist[i - gap]:
12                 alist[i], alist[i - gap] = alist[i - gap], alist[i]
13                 i -= gap
14 
15         # 缩短gap步长,不一定必须折半
16         gap //= 2
17 
18 
19 if __name__ == "__main__":
20     li = [54, 26, 93, 17, 77, 31, 44, 55, 20]
21     shell_sort(li)
22     print(li)

 

 

 

 5、快速排序  Quick Sort

(1)选择一个关键值,通过一趟排序将要排序的数据分割成独立的两部分,其中一部分的所有数据都比另外一部分的所有数据小,然后再按此方法对这两部分数据分别进行快速排序,整个排序过程递归进行。

(2)图示

  <1> 详细拆解

   

  <2> 代码中的low和high

(3)代码

 1 def quick_sort(alist, first, last):
 2     """快速排序"""
 3     if first < last:
 4 
 5         # 设置中间值,作为基数:最终比他小的元素放在左边,比他大的元素放在右边
 6         # 此处就是取的第一个元素作为基数
 7         mid_value = alist[first]
 8 
 9         # 设置高、低两个标记位
10         low,high = first,last
11 
12         # 每一次外循环完成的是:一次高、低位的交换(和中间值比较后)
13         while low < high:
14 
15             # Step1 high左移:如果高标记位所在的值大于等于中间值,则继续左移
16             while low < high and alist[high] >= mid_value:
17                 high -= 1
18 
19             # Step2 当高标记位的值小于中间值时:标记位不再移动,同时,将高标记位的值放到低标记位
20             alist[low] = alist[high]
21 
22             # Step3 low右移:如果低记位所在的值小于于中间值,则继续左移
23             # 注意:此处是没有等于mid_value的情况,目的是将同等大小情况的值,都放到一边,这是一种规范
24             while low < high and alist[low] < mid_value:
25                 low += 1
26 
27             # Step4 当低标记位的值大于中间值时:标记位不再移动,同时,将低标记位的值放到高标记位
28             alist[high] = alist[low]
29 
30         # 从循环退出时,low>=high,将中间值直接放到low位
31         # 此时,mid_value的左边都是比他小的,右边都是比他大的
32         alist[low] = mid_value
33 
34         # 递归一 :对中间值左边的列表执行快速排序
35         quick_sort(alist, first, low-1)
36 
37         # 递归二 :对中间值右边的列表排序
38         quick_sort(alist, low+1, last)
39 
40     return alist
41 
42 
43 if __name__ == "__main__":
44     li = [54, 26, 93, 17, 77, 31, 44, 55, 20]
45     quick_sort(li, 0, len(li)-1)
46     print(li)

 

 

 

6、归并排序

(1)先递归分解数组,子数组之间比较后合并成有序数组,新合并的子数组之间再递归比较,直到最终合并成唯一一个新的有序数组。

(2)步骤:先将原数组分解最小之后,然后合并两个有序数组(依次重新合并,成为最终唯一的有序数组)。其中,每次两个小分组比较时,基本思路是比较两个数组的最前面的数,谁小就先取谁,取了后相应的指针就往后移一位。然后再比较,直至一个数组为空,最后把另一个数组的剩余部分复制过来即可。

(3)图示

    

(4)代码

 1 def merge_sort(alist):
 2     """归并排序"""
 3     n = len(alist)
 4 
 5     # 个数大于一时进行比较,否则直接返回
 6     if n > 1:
 7 
 8         # ---------- 递归分解数组 ----------
 9 
10         # 每次采用折半的方式进行拆分,直到拆成一个元素的数组
11         mid = n//2
12 
13         # left:左半部分,递归,采用归并排序后形成的有序的新的列表
14         left_li = merge_sort(alist[:mid])
15 
16         # right:右半部分,递归,采用归并排序后形成的有序的新的列表
17         right_li = merge_sort(alist[mid:])
18 
19 
20 
21         # ---------- 递归排序子序列并且合并 ----------
22 
23         # 将两个有序的子序列合并为一个新的整体
24         # 左、右指针,用于依次取出左右两个数组的每个元素进行比较
25         left_pointer, right_pointer = 0, 0
26 
27         # 存放合并后的有序数组
28         result = []
29 
30         # 循环,直到穷尽左、右其中一个数组
31         while left_pointer < len(left_li) and right_pointer < len(right_li):
32 
33             # 比较左右子序列,把小的依次放到result里,并且正确的移动相应的指针
34             if left_li[left_pointer] <= right_li[right_pointer]:
35                 result.append(left_li[left_pointer])
36                 left_pointer += 1
37             else:
38                 result.append(right_li[right_pointer])
39                 right_pointer += 1
40 
41         # 跳出循环,不论左、右序列是否有剩余的元素,直接添加到结果最后
42         # 注意:python的数组切片时,如果越界,返回空数组
43         result += left_li[left_pointer:]
44         result += right_li[right_pointer:]
45         
46         # 返回新的有序数组
47         return result
48     
49     return alist
50 
51 
52 if __name__ == "__main__":
53     li = [54, 26, 93, 17, 77, 31, 44, 55, 20]
54     sorted_li = merge_sort(li)
55     print(sorted_li)

 

 

 

 

三、常见排序算法效率比较

 

 

 

 

四、搜索

 

1、二分法查找、折半查找

(1)针对有序序列,选择中间点,比较查找目标在左右哪个区间,然后递归在该区间查找目标,直到该目标和中间点的值相同(查找到了),否则无结果(没查找到)

(2)代码实现

  <1> 递归版本

 1 def binary_search(alist, item):
 2     """二分查找,递归"""
 3     n = len(alist)
 4     if n > 0:
 5 
 6         # 每次递归,设置中间角标
 7         mid = n//2
 8 
 9         # Section1 查找的值等于中间值,查找成功!
10         if alist[mid] == item:
11             return '找到了:' + str(item)
12 
13         # Section2 查找的值小于中间值,递归查找中间值左边的子序列
14         elif item < alist[mid]:
15             return binary_search(alist[:mid], item)
16 
17         # Section3 查找的值大于中间值,递归查找中间值右边的子序列
18         else:
19             return binary_search(alist[mid+1:], item)
20 
21     # 传入序列元素为空,结束,未查找到!
22     return '没有找到:' + str(item)
23 
24 
25 if __name__ == "__main__":
26     li = [17, 20, 26, 31, 44, 54, 55, 77, 93]
27     print(binary_search(li, 55))
28     print(binary_search(li, 100))

  <2> 非递归

 1 def binary_search_2(alist, item):
 2     """二分查找, 非递归"""
 3     n = len(alist)
 4     first = 0
 5     last = n-1
 6 
 7     # 更改查找的左右区间,直到左指针超过了右指针,循环停止
 8     while first <= last:
 9         
10         # 取出中间值坐标
11         mid = (first + last)//2
12         if alist[mid] == item:
13             return "找到了:" + str(item)
14         elif item < alist[mid]:
15             # 右指针设置为中间值左边第一个
16             last = mid - 1
17         else:
18             # 左指针设置为中间值右边第一个
19             first = mid + 1
20             
21     return "未找到:" + str(item)
22 
23 
24 if __name__ == "__main__":
25     
26     li = [17, 20, 26, 31, 44, 54, 55, 77, 93]
27     print(binary_search_2(li, 55))
28     print(binary_search_2(li, 100))

 

 

 

 

五、树与树算法

 

1、树的特质

(1)完全二叉树:若设二叉树的高度为h,除第 h 层外,其它各层 (1~h-1) 的结点数都达到最大个数,第h层有叶子结点,并且叶子结点都是从左到右依次排布。

(2)满二叉树:除了叶结点外每一个结点都有左右子叶且叶子结点都处在最底层的二叉树。

(3)平衡二叉树:又被称为AVL树(区别于AVL算法),它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。

(3)特质:

  <1>:  在二叉树的第i层上至多有2^(i-1)个结点(i>0)
  <2>: 深度为k的二叉树至多有2^k - 1个结点(k>0)
  <3>: 对于任意一棵二叉树,如果其叶结点数为N0,而度数为2的结点总数为N2,则N0=N2+1;
  <4>: 具有n个结点的完全二叉树的深度必为 log2(n+1)
  <5>: 对完全二叉树,若从上至下、从左至右编号,则编号为i 的结点,其左孩子编号必为2i,其右孩子编号必为2i+1;其双亲的编号必为i/2(i=1 时为根,除外)

(4)几种遍历方式:广度遍历,先序遍历(根左右),中序遍历(左根右),后续遍历(左右根)

(5)生成树(包括添加元素、广度遍历)时,使用队列,来存储每个待处理节点,通过队列的pop和append,逐步完成整二叉树的操作。 

   

  2、代码实现

 1 class Node(object):
 2     """节点"""
 3     def __init__(self, item):
 4         self.elem = item
 5         self.lchild = None
 6         self.rchild = None
 7 
 8 class Tree(object):
 9     """二叉树"""
10     def __init__(self):
11         self.root = None
12 
13     def add(self, item):
14         """添加节点"""
15         node = Node(item)
16         if self.root is None:
17             self.root = node
18             return
19 
20         # 取出根节点放到队列中
21         queue = [self.root]
22 
23         # 只要队列不为空,就一直从队列中取出节点处理
24         while queue:
25 
26             # 取出当前要处理节点
27             cur_node = queue.pop(0)
28 
29             # 如果左节点不存在,直接设置item到左节点
30             if cur_node.lchild is None:
31                 cur_node.lchild = node
32                 return
33             else:
34                 # 如果左节点存在,将其加入队列中,作为"待处理"的节点
35                 queue.append(cur_node.lchild)
36 
37             # 同上,判断右节点
38             if cur_node.rchild is None:
39                 cur_node.rchild = node
40                 return
41             else:
42                 queue.append(cur_node.rchild)
43 
44     def breadth_travel(self):
45         """广度遍历"""
46         if self.root is None:
47             return
48         queue = [self.root]
49         while queue:
50             cur_node = queue.pop(0)
51             print(cur_node.elem, end=" ")
52             if cur_node.lchild is not None:
53                 queue.append(cur_node.lchild)
54             if cur_node.rchild is not None:
55                 queue.append(cur_node.rchild)
56 
57     def preorder(self, node):
58         """先序遍历"""
59         if node is None:
60             return
61         print(node.elem, end=" ")
62         self.preorder(node.lchild)
63         self.preorder(node.rchild)
64 
65     def inorder(self, node):
66         """中序遍历"""
67         if node is None:
68             return
69         self.inorder(node.lchild)
70         print(node.elem, end=" ")
71         self.inorder(node.rchild)
72 
73     def postorder(self, node):
74         """后序遍历"""
75         if node is None:
76             return
77         self.postorder(node.lchild)
78         self.postorder(node.rchild)
79         print(node.elem, end=" ")
80 
81 
82 if __name__ == "__main__":
83 
84     tree = Tree()
85     print("\n创建数据结构 0 - 9 数字")
86     for index in range(10):
87         tree.add(index)
88     print("\n广度遍历:")
89     tree.breadth_travel()
90     print("\n前序遍历:")
91     tree.preorder(tree.root)
92     print("\n中序遍历:")
93     tree.inorder(tree.root)
94     print("\n后序遍历:")
95     tree.postorder(tree.root)
96     print("\n")

 

推荐阅读