最小的K个数

最新推荐文章于 2024-09-27 09:30:58 发布

WYQ_XQ

最新推荐文章于 2024-09-27 09:30:58 发布

阅读量291

点赞数

分类专栏：算法文章标签：堆排序最小的K个数基础算法

本文链接：https://blog.csdn.net/u014041033/article/details/51821166

版权

算法专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种基于堆排序原理来寻找数组中最小的K个数的方法。通过实现HeapSort和AdjustHeap函数，该算法可以高效地找到指定数量的最小元素，并返回这些元素组成的列表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<Integer>();
if(input==null||input.length==0||k>input.length){
return list;
}
HeapSort(input,k);

for(int x=k;x<input.length;x++){
if(input[x]<input[0]){
input[0]=input[x];
AdjustHeap(input,0,k);
}
}
for(int x=0;x<k;x++){
list.add(input[x]);
}
return list;
}
public void AdjustHeap(int a[],int i,int length){
int child;
int temp;
for(;i*2+1<length;i=child){
child = 2*i+1;
temp=a[child];
if(child<length-1&&a[child]<a[child+1]){
child++;
}
if(a[i]<a[child]){
temp=a[child];
a[child]=a[i];
a[i] = temp;
}
else break;
}
}
public void HeapSort(int a[],int length){
for(int i=length/2-1;i>=0;i--){
AdjustHeap(a,i,length);
}
}

}

题目来源：最小的k个数

参考：堆排序原理