码迷,mamicode.com
首页 > 编程语言 > 详细

找出数组中前K小的值&最小堆

时间:2015-08-09 14:13:17      阅读:183      评论:0      收藏:0      [点我收藏+]

标签:java

题目描述

输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
代码
 public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        
        if (input.length==0||k>input.length) {
			return new ArrayList<>();
		}
		ArrayList<Integer> list = new ArrayList<>();
		PriorityQueue<Integer> priorityQueue = new PriorityQueue<>();
		for (int integer : input) {
			priorityQueue.add(integer);
		}
		
		for (int i = 0; i < k; i++) {
			list.add(priorityQueue.poll());
		}
		return list;
    }


    在平时的编程工作中似乎很少碰到PriorityQueue(优先队列) ,故很多人一开始看到优先队列的时候还会有点迷惑。优先队列本质上就是一个最小堆。如果是最大堆,则二叉树的顶点是保存的最大值,最小堆则保存的最小值。最小堆每一个节点都是以该节点为根的子树中的最小值。

    下面是一个典型的优先队列的结构图:

技术分享

 

    它的每个父节点都比两个子节点要小,但是整个数组又不是完全顺序的。

    

版权声明:本文为博主原创文章,未经博主允许不得转载。

找出数组中前K小的值&最小堆

标签:java

原文地址:http://blog.csdn.net/qingchunweiliang/article/details/47375177

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!