您的位置:首页 > 健康 > 美食 > 最小k个数

最小k个数

2024/10/6 12:35:03 来源:https://blog.csdn.net/m0_63703622/article/details/141760712  浏览:    关键词:最小k个数

力扣(LeetCode)----前k个最小数

1 题目描述

设计一个算法,找出数组中最小的k个数。以任意顺序返回这k个数均可。

示例:

输入: arr = [1,3,5,7,2,4,6,8], k = 4
输出: [1,2,3,4]

提示:

  • 0 <= len(arr) <= 100000

  • 0 <= k <= min(100000, len(arr))

这个题如果用普通的排序操作其实很简单,直接用冒泡把所有数字从小到大排序就行,但如果数据量很大呢,大到内存存不下的话,就会导致冒泡这类排序的时间复杂度太大导致无法实现。

这时我们就需要利用到堆。

2 解题思路

 

假如给了一个N个元素大的数组,需要去取得这个数组里面的前5个最小的数。

这时候我们可以将arr内前五个元素取出来去建立一个大堆,因为要取前几个最小的数,所以我们建一个大根堆。

通过循环向下调整算法去将它调整为大根堆。


void AdjustDown(int *a,int k,int root)
{int parent=root;int child=parent*2+1;while(child<k){if(child+1<k&&a[child]<a[child+1]){child++;}if(a[child]>a[parent]){int temp=a[parent];a[parent]=a[child];a[child]=temp;parent=child;child=parent*2+1;}else{break;}}
}

然后再将数组中后面没用来建堆的数来和大堆的头去比较,小于堆顶的直接入堆。

然后在进行向下调整。

这样我们就把小的数字放到最下面,重复这个步骤就能把这N个最小的数字给放到这个堆里面去。

3 完整代码

/*** Note: The returned array must be malloced, assume caller calls free().*/void AdjustDown(int *a,int k,int root)
{int parent=root;int child=parent*2+1;while(child<k){if(child+1<k&&a[child]<a[child+1]){child++;}if(a[child]>a[parent]){int temp=a[parent];a[parent]=a[child];a[child]=temp;parent=child;child=parent*2+1;}else{break;}}
}int* smallestK(int* arr, int arrSize, int k, int* returnSize){*returnSize=k;if(k==0) return NULL;int *a=(int*)malloc(sizeof(int)*k);memcpy(a,arr,sizeof(int)*k);for(int i=(k-1-1)/2;i>=0;i--){AdjustDown(a,k,i);}for(int i=k;i<arrSize;i++){if(a[0]>arr[i]){a[0]=arr[i];AdjustDown(a,k,0);}}return a;
}

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com