标签:arr old 21 IntHeap int 个数 func return Leetcode
输入整数数组 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]
思想:实现一个大根堆,通过维护大根堆,实现升序排序来找到最小的k个数。
import ( "container/heap" ) type IntHeap []int func(h IntHeap) Len() int{ return len(h) } func(h IntHeap) Less(i,j int) bool{ return h[i]>h[j] } func(h IntHeap) Swap(i,j int){ h[i],h[j]=h[j],h[i] } func(h *IntHeap) Push(x interface{}){ *h = append(*h,x.(int)) } func(h *IntHeap) Pop() interface{}{ old := *h n:=len(old) x := old[n-1] *h = old[0:n-1] return x } func getLeastNumbers(arr []int, k int) []int { if k ==0{ return []int{} } h := make(IntHeap,k) hp := &h copy(h,IntHeap(arr[:k+1])) heap.Init(hp) for i:=k;i<len(arr);i++{ if arr[i]<h[0]{ heap.Pop(hp) heap.Push(hp,arr[i]) } } return h }
这里要注意大根堆的实现方法,以及使用方式。
题目来源:https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof
标签:arr,old,21,IntHeap,int,个数,func,return,Leetcode 来源: https://www.cnblogs.com/End1ess/p/15485368.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。