标签：tmp arr int 堆排序 len li golang func leetcode

package main

import (
	"fmt"
)

/*
输入整数数组 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]
*/

func main() {
	fmt.Println(getLeastNumbers([]int{2, 3, 1, 5, 9, 4, 1, 3}, 3))
}

func getLeastNumbers(arr []int, k int) []int {
	heapSort(arr, k)
	return arr[len(arr)-k:]
}
func heapSort(li []int, k int) {
	// 构建小根堆, 从最后一个非叶子节点进行构建
	tmp := (len(li) - 1 - 1) / 2  // 获取最后一个非叶子节点
	for i := tmp; i >= 0; i-- {
		sift(li, i, len(li)-1)
	}

	// 堆构建完成，开始进行排序（排序的过程本质上就是构建堆的过程）
	// 注意：因为我们只需要前k小的数，所以我们只需要排完前k个就可以了
	for i := len(li)-1; i >= len(li)-k; i-- {
		li[0], li[i] = li[i], li[0]
		sift(li, 0, i-1)
	}

}
func sift(li []int, low, high int) {
	tmp := li[low]    // 暂存当前节点数据
	left := 2*low + 1 // 左孩子
	for left <= high {  // 只要有左孩子就遍历
		if left+1 <= high && li[left+1] < li[left] { // 如果有右孩子，而且右孩子比左孩子小
			left += 1
		}
		if li[left] < tmp {
			li[low] = li[left]
			low = left // 往下看一层
			left = 2*low + 1  // 同时修改左孩子
		}else {
			break
		}
	}
	li[low] = tmp  // 把tmp归位
}

标签：tmp,arr,int,堆排序,len,li,golang,func,leetcode
来源： https://www.cnblogs.com/mayanan/p/16455315.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。