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

refactor: Replace 'poll' with 'pop' in Heap (#416)

Browse Source
This commit is contained in:
Yudong Jin
2023-03-13 22:31:05 +08:00
committed by GitHub
parent 8aebbaad21
commit 28aacccf44
27 changed files with 91 additions and 152 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
codes/python/chapter_heap/my_heap.py
View File

@ -67,7 +67,7 @@ class MaxHeap:
# 循环向上堆化
i = p
def poll(self) -> int:
def pop(self) -> int:
""" 元素出堆 """
# 判空处理
assert not self.is_empty()
@ -102,18 +102,6 @@ class MaxHeap:
print_heap(self.max_heap)
def test_push(max_heap: MaxHeap, val: int):
max_heap.push(val) # 元素入堆
print(f"\n添加元素 {val} 后\n")
max_heap.print()
def test_poll(max_heap: MaxHeap):
val = max_heap.poll() # 堆顶元素出堆
print(f"\n出堆元素为 {val}\n")
max_heap.print()
""" Driver Code """
if __name__ == "__main__":
# 初始化大顶堆
@ -132,7 +120,7 @@ if __name__ == "__main__":
max_heap.print()
# 堆顶元素出堆
peek = max_heap.poll()
peek = max_heap.pop()
print(f"\n堆顶元素 {peek} 出堆后")
max_heap.print()