最小的k个数1 堆排序实现
// 使用堆排序实现 其时间复杂度为O(nlgn)
private static void buildMaxHeap(int[] input, int end)
{
// 从非叶子节点开始进行
for (int i = (end - 1) / 2; i >= 0; i--)
{
// 当前节点 cur的字节点位cur*2+1&cur*2+2
int cur = i;
// cur的右孩子存在
while (cur * 2 + 1 < end)
{
int bigIndex = cur * 2 + 1;
int rightOfCur = cur * 2 + 2;
// 找出孩子节点中的最大值索引
if (input[bigIndex] < input[rightOfCur])
{
bigIndex++;
}
// input[cur]节点与input[bigIndex] 进行比较进
if (input[cur] < input[bigIndex])
{
int temp = input[cur];
input[cur] = input[bigIndex];
input[bigIndex] = temp;
// 尤其重要
cur = bigIndex;
}
else
{
break;
}
}
}
}
private static int[] minKOfNums(int[] nums, int k)
{
// 特殊值考虑
if (nums == null || nums.length < k || k <= 0)
{
return null;
}
// 声明堆
int[] maxHeap = new int[k];
// 初始化
for (int i = 0; i < k; i++)
{
maxHeap[i] = nums[i];
}
// 建成最大堆
buildMaxHeap(maxHeap, maxHeap.length - 1);
// 进行后续的比较
for (int i = k; i < nums.length; i++)
{
if (maxHeap[0] > nums[i])
{
maxHeap[0] = nums[i];
// 重新建maxHeap
buildMaxHeap(maxHeap, maxHeap.length - 1);
}
}
return maxHeap;
}