zoukankan      html  css  js  c++  java
  • 剑指offer 29.最小的K个数

    剑指offer 29.最小的K个数

    题目

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

    思路

    几种方法,我一开始想到的是直接快排函数,然后输出k个就行,之后改用优先队列建堆,大小控制为k,大于这个数量就删除堆顶(最大值),最后只剩k个,输出到ans当中即可。

    代码

      public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
    
        ArrayList<Integer> ans = new ArrayList<>();
        if (input == null || k <= 0 || k > input.length) {
          return ans;
        }
        Queue<Integer> queue = new PriorityQueue<>((o1, o2) -> o2 - o1);
    
        for (int i = 0; i < input.length; i++) {
    
          if (queue.size() < k) {
            queue.add(input[i]);
          } else {
            if (input[i] < queue.peek()) {
              queue.remove();
              queue.add(input[i]);
            }
          }
        }
        while (!queue.isEmpty()) {
          ans.add(queue.remove());
        }
        return ans;
      }
    
  • 相关阅读:
    基于JavaMail开发邮件发送器工具类
    微服务架构
    图像识别
    Manacher
    左偏树
    虚树
    动态树(Link-Cut Tree)
    Splay
    扩展中国剩余定理(EXCRT)
    K-D树
  • 原文地址:https://www.cnblogs.com/blogxjc/p/12405351.html
Copyright © 2011-2022 走看看