输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。

来源:互联网 发布:纯牛奶 知乎 编辑:程序博客网 时间:2024/05/29 16:15

如题:剑指offer
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
使用堆排序在大批量的排序更快,复杂度NlogM

public class Solution {    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {       ArrayList<Integer> result = new ArrayList<Integer>();       int length = input.lenght;       if(k>length||k==0){         return result;         }       PriorityQueue<Integer> maxHeap = new  PriorityQueue<Integer>(k,new Comparator<Integer>(){          public int compare(Integer o1,Integer o2){                  return o2.compareTo(o1);         }       });       for(int i = 0;i<length;i++){          if(maxHeap.size()!=k){              maxHeap.offer(input[i]);        }else if(maxHeap.peek()>input[i]){             Integer temp = maxHeap.poll();             temp = null;             maxHeap.offer(input[i]);             }           }           for(Integer integer:maxHeap) {                 result.add(integer);      }      return result;    }    }   } }
0 0
原创粉丝点击