刷题笔记:Java中堆的使用(最小的K个数)

最小的K个数

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

import java.util.*;
public class Solution {
    public ArrayList GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList res = new ArrayList();
        if(input.length que = new PriorityQueue(new Comparator() {
              public int compare(Integer e1, Integer e2) {
                  return e2.compareTo(e1);
              }
        });
        for(int i =0 ;i0){
            res.add(que.poll());
        }
        return res;
    }
}

你可能感兴趣的:(刷题笔记)