剑指JZ29—最小的k个数（小根堆）

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。

解题思路

这题属于比较简单的题目，可以利用小根堆来解决，首先时建堆过程，然后每次找出堆顶元素，并进行调整

def HeapChange(nums, k):
    m = len(nums) - 1
    i = 2 * k + 1
    while i <= m:
        while i < m and nums[i] > nums[i + 1]:
            i += 1
        if nums[k] > nums[i]:
            nums[k], nums[i] = nums[i], nums[k]
            k = i
            i = 2 * k + 1
        else:
            break

def Find_Min_K(nums, k):
    m = len(nums) // 2
    res = []
    for i in range(0, m):
        HeapChange(nums, m - i - 1)
    for i in range(k):
        res.append(nums[0])
        nums[0], nums[len(nums) - 1] = nums[len(nums) - 1], nums[0]
        nums.pop()
        HeapChange(nums, 0)
    return res

nums = [4, 5, 1, 6, 2, 7, 3, 8]
print(Find_Min_K(nums, 4))