标签:大顶 arr int res 个数 最小 40 queue
文章目录
剑指 Offer 40. 最小的k个数
输入整数数组 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]
限制:
0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000
解法一:排序
对数组快速排序,然后返回前k
个数即可。
时间复杂度:
O
(
n
l
o
g
n
)
O(nlogn)
O(nlogn)
Java代码
class Solution {
public int[] getLeastNumbers(int[] arr, int k) {
int[] res = new int[k];
Arrays.sort(arr);
for(int i = 0;i < k;i++){
res[i] = arr[i];
}
return res;
}
}
解法二:优先队列
用一个大顶堆
,遍历一次数组,每次都将元素加入大顶堆
,如果此时大顶堆
的元素大于k
个了,则把堆顶的元素去掉,即保证大顶堆中保留的一直是最小的k个数
。
注:
这里用优先队列的效果还没有上面直接快速排序好,在这里用优先队列就是简要介绍一下他的用法,因为后面有好几道题都需要用优先队列。
Java代码
class Solution {
public int[] getLeastNumbers(int[] arr, int k) {
int[] res = new int[k];
//PriorityQueue是Queue的一个实现类
//默认是小顶堆,建立大顶堆则需要构造时传自定义的Comparator参数
Queue<Integer> queue = new PriorityQueue<>(new Comparator<Integer>(){
@Override
public int compare(Integer o1,Integer o2){
return o2 - o1;//从大到小排
}
});
//大顶堆中保留最小的k个数
for(int i = 0;i < arr.length;i++){
queue.add(arr[i]);
if(queue.size() > k) queue.poll();
}
for(int i = 0;i < k;i++){
res[i] = queue.poll();
}
return res;
}
}
标签:大顶,arr,int,res,个数,最小,40,queue 来源: https://blog.csdn.net/YouMing_Li/article/details/114527881
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。