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

feat: add top_k.c and refactor top_k.js (#889)

Browse Source 
* Add top_k.c based on my_heap.c

* Improve the implementation of top_k.js

* Add a comment to top_k
This commit is contained in:
Yudong Jin
2023-10-26 02:54:19 +08:00
committed by GitHub
parent 9f4076d1c1
commit 7822bf9cd4
13 changed files with 190 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
codes/python/chapter_heap/top_k.py
View File

@ -15,6 +15,7 @@ import heapq
def top_k_heap(nums: list[int], k: int) -> list[int]:
"""基于堆查找数组中最大的 k 个元素"""
# 初始化小顶堆
heap = []
# 将数组的前 k 个元素入堆
for i in range(k):