剑指 Offer 40. 最小的k个数（golang版）

• 题目描述

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

示例 1：
输入：arr = [3,2,1], k = 2
输出：[1,2] 或者 [2,1]

示例 2：
输入：arr = [0,1,2,1], k = 1
输出：[0]
 
限制：
0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

• 思路
  1、堆排序，取最k个值
• 代码

func getLeastNumbers(arr []int, k int) []int {
	for i:=len(arr); i>len(arr)-k; i-- {
		heap(arr[:i])
		arr[0],arr[i-1] = arr[i-1],arr[0]
	}
	return arr[len(arr)-k:]
}

func heap(arr []int){
	len := len(arr)
	for i:=len/2-1; i>=0; i-- {
		if 2*i+1<len && arr[i] > arr[2*i+1] {
			arr[i],arr[2*i+1] = arr[2*i+1],arr[i]
		}
		if 2*i+2<len && arr[i] > arr[2*i+2] {
			arr[i],arr[2*i+2] = arr[2*i+2],arr[i]
		}
	}
}