面试题30、输入n个整数,找出其中最小的k个数,例如输入4,5,1,6,2,7,3,8这个8个数字,则最小的4个数字是1,2,3,4
import java.util.ArrayList;
import java.util.Comparator;
import java.util.PriorityQueue;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> result = new ArrayList<Integer>();
if(k > input.length || k<=0){
return result;
}
//定义排列的顺序
Comparator<Integer> cmp;
cmp = new Comparator<Integer>(){
public int compare(Integer o1, Integer o2){
return o2-o1;
}
};
//java中的最大堆结构
PriorityQueue<Integer> qi = new PriorityQueue<Integer>(k,cmp);
//添加k个元素,按照从小到大的顺序进行排列,能够逆序
int i = 0;
for(;i<k;i++){
qi.add(input[i]);
}
// System.out.println("queue输出"+qi.size());
// while(!qi.isEmpty()){
// System.out.print(qi.poll()+" ");
// }
for(;i<input.length ;i++){
if(input[i] < qi.peek()){
qi.poll();
qi.add(input[i]);
}
}
while(!qi.isEmpty()){
result.add(qi.poll());
}
return result;
}
}
《剑指offer》中和大数据相关的题目
最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...