Skip to content

Latest commit

 

History

History
67 lines (57 loc) · 2.17 KB

40.最小的K个数.md

File metadata and controls

67 lines (57 loc) · 2.17 KB

40.最小的k个数

题目描述

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

思路 & 解答

下面方法的思路比较简单,但是比较取巧,直接借助了大顶堆,也就是上面的元素是最大的。如果里面个数超过k的话,则弹出堆顶的元素。 这里不展开最大堆和最小堆的具体讲解,什么是最大堆/最小堆呢?

  • 大/小堆树是完全二叉树
  • 大/小堆树的每一个节点不小于/不大于其孩子节点。

这道题我们用的是最大堆,也就是最上面的元素是最大的,如果个数超过 k 的话,那么就把堆顶的元素弹出去,也就是优先将最大的弹出。

代码如下:

import java.util.ArrayList;
import java.util.PriorityQueue;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        if (input == null || k > input.length || k <= 0) {
            return new ArrayList<>();
        }
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>((o1, o2) -> o2 - o1);
        for (int num : input) {
            priorityQueue.add(num);
            if (priorityQueue.size() > k) {
                priorityQueue.poll();
            }
        }
        return new ArrayList<>(priorityQueue);
    }
}

C++ 代码实现如下:

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> ret;
        if (k == 0 || k > input.size()) return ret;
        priority_queue<int, vector<int>> pmyQueue;
        for (const int val : input) {
            if (pmyQueue.size() < k) {
                pmyQueue.push(val);
            } else {
                if (val < pmyQueue.top()) {
                    pmyQueue.pop();
                    pmyQueue.push(val);
                }
            }
        }

        while (!pmyQueue.empty()) {
            ret.push_back(pmyQueue.top());
            pmyQueue.pop();
        }
        return ret;
    }
};
  • 时间复杂度:O(nlogk), 插入大小为k的大根堆的时间复杂度为O(logk), n`个元素
  • 空间复杂度:O(k)