java 求最大、最小K个数，代码

最新推荐文章于 2021-11-21 12:07:42 发布

NWPU_Frank.liu

最新推荐文章于 2021-11-21 12:07:42 发布

阅读量229

点赞数

本文链接：https://blog.csdn.net/NWPU_LIUBO/article/details/104467834

版权

求最大K个数，构造小顶堆。

求最小K个数，构造大顶堆。

代码如下，看注释

package javaproject;

import java.awt.List;
import java.util.*;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Scanner;


import java.util.Scanner;
import java.util.Set;
 
public class Main{

	    public static ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
	        ArrayList<Integer> result = new ArrayList<Integer>();
	        int length = input.length;
	        if(k > length || k == 0){
	            return result;
	        }
	     
	        PriorityQueue<Integer> maxHeap = new PriorityQueue<Integer>(k, new Comparator<Integer>() {

	            @Override
	            public int compare(Integer o1, Integer o2) {
	                return o1-o2; // 构造小顶堆 （默认）   求最大K个数
	              //return o2-o1;    构造大顶堆                   求最小K个数
	            }
	        });

	        for (int i = 0; i < length; i++) {
	            if (maxHeap.size() != k) {
	                maxHeap.offer(input[i]);
	            } else if (input[i] > maxHeap.peek())  //此处注意大于小于号： >代表求最大K个数，  <代表求最小K个数
	            {
	                Integer temp = maxHeap.poll();
	                temp = null;
	                maxHeap.offer(input[i]);
	            }
	        }
	        for (Integer integer : maxHeap) {
	            result.add(integer);
	        }
	        return result;
	    }
}