algorithm/hello-algo
1
0
Fork 0
mirror of https://github.com/krahets/hello-algo.git synced 2025-11-02 12:58:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add python code of Heap and Graph to docs.

Browse Source
This commit is contained in:
krahets
2023-02-23 20:23:49 +08:00
parent 1f4dba4845
commit 73c8920c6b
5 changed files with 64 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/chapter_graph/graph_traversal.md
View File

@ -43,7 +43,7 @@ BFS 常借助「队列」来实现。队列具有“先入先出”的性质，
=== "Python"
```python title="graph_bfs.py"
[class]{}-[func]{graph_bfs}
```
=== "Go"
@ -160,7 +160,9 @@ BFS 常借助「队列」来实现。队列具有“先入先出”的性质，
=== "Python"
```python title="graph_dfs.py"
[class]{}-[func]{dfs}
[class]{}-[func]{graph_dfs}
```
=== "Go"

46
docs/chapter_heap/heap.md
View File

@ -123,7 +123,41 @@ comments: true
=== "Python"
```python title="heap.py"
# 初始化小顶堆
min_heap, flag = [], 1
# 初始化大顶堆
max_heap, flag = [], -1
# Python 的 heapq 模块默认实现小顶堆
# 考虑将“元素取负”后再入堆，这样就可以将大小关系颠倒，从而实现大顶堆
# 在本示例中，flag = 1 时对应小顶堆，flag = -1 时对应大顶堆
""" 元素入堆 """
heapq.heappush(max_heap, flag * 1)
heapq.heappush(max_heap, flag * 3)
heapq.heappush(max_heap, flag * 2)
heapq.heappush(max_heap, flag * 5)
heapq.heappush(max_heap, flag * 4)
""" 获取堆顶元素 """
peek = flag * max_heap[0] # 5
""" 堆顶元素出堆 """
# 出堆元素会形成一个从大到小的序列
val = flag * heapq.heappop(max_heap) # 5
val = flag * heapq.heappop(max_heap) # 4
val = flag * heapq.heappop(max_heap) # 3
val = flag * heapq.heappop(max_heap) # 2
val = flag * heapq.heappop(max_heap) # 1
""" 获取堆大小 """
size = len(max_heap)
""" 判断堆是否为空 """
is_empty = not max_heap
""" 输入列表并建堆 """
min_heap = [1, 3, 2, 5, 4]
heapq.heapify(min_heap)
```
=== "Go"
@ -311,7 +345,11 @@ comments: true
=== "Python"
```python title="my_heap.py"
[class]{MaxHeap}-[func]{left}
[class]{MaxHeap}-[func]{right}
[class]{MaxHeap}-[func]{parent}
```
=== "Go"
@ -403,7 +441,7 @@ comments: true
=== "Python"
```python title="my_heap.py"
[class]{MaxHeap}-[func]{peek}
```
=== "Go"
@ -493,7 +531,9 @@ comments: true
=== "Python"
```python title="my_heap.py"
[class]{MaxHeap}-[func]{push}
[class]{MaxHeap}-[func]{sift_up}
```
=== "Go"
@ -613,7 +653,9 @@ comments: true
=== "Python"
```python title="my_heap.py"
[class]{MaxHeap}-[func]{poll}
[class]{MaxHeap}-[func]{sift_down}
```
=== "Go"
@ -693,7 +735,7 @@ comments: true
=== "Python"
```python title="my_heap.py"
[class]{MaxHeap}-[func]{__init__}
```
=== "Go"