【牛客面试必刷TOP101】Day26.（X）BM41 输出二叉树的右视图和BM46 最小的K个数

作者简介：大家好，我是未央；

博客首页：未央.303

系列专栏：牛客面试必刷TOP101

每日一句：人的一生，可以有所作为的时机只有一次，那就是现在！！！！

文章目录

• 前言
• 一、BM41 输出二叉树的右视图
• 题目描述
• 题目解析
• 二、BM46 最小的K个数
• 题目描述
• 题目解析
• 总结

---

前言

---

一、BM41 输出二叉树的右视图

题目描述

描述：

请根据二叉树的前序遍历，中序遍历恢复二叉树，并打印出二叉树的右视图

---

数据范围： 0≤n≤10000
要求： 空间复杂度 O(n)，时间复杂度 O(n)

---

举例说明：

如输入[1,2,4,5,3],[4,2,5,1,3]时，通过前序遍历的结果[1,2,4,5,3]和中序遍历的结果[4,2,5,1,3]可重建出以下二叉树：

所以对应的输出为[1,3,5]。

示例1:

---

题目解析

---

---

二、BM46 最小的K个数

 题目描述

描述：

给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。

---

数据范围：0≤k,n≤10000，数组中每个数的大小0≤val≤1000

要求：空间复杂度 O(n)，时间复杂度 O(nlogk)

示例1:

---

示例2:

---

示例3:

---

题目解析

知识点：优先队列

优先队列即PriorityQueue，是一种内置的机遇堆排序的容器，分为大顶堆与小顶堆，大顶堆的堆顶为最大元素，其余更小的元素在堆下方，小顶堆与其刚好相反。且因为容器内部的次序基于堆排序，因此每次插入元素时间复杂度都是O(log2n)，而每次取出堆顶元素都是直接取出。

思路：

要找到最小的k个元素，只需要准备k个数字，之后每次遇到一个数字能够快速的与这k个数字中最大的值比较，每次将最大的值替换掉，那么最后剩余的就是k个最小的数字了。

如何快速比较k个数字的最大值，并每次替换成较小的新数字呢？

我们可以考虑使用优先队列（大根堆），只要限制堆的大小为k，那么堆顶就是k个数字的中最大值，如果需要替换，将这个最大值拿出，加入新的元素就好了。

//较小元素入堆
if(q.peek() > input[i]){  
    q.poll();
    q.offer(input[i]);
}

---

具体做法：

• step 1：利用input数组中前k个元素，构建一个大小为k的大顶堆，堆顶为这k个元素的最大值。
• step 2：对于后续的元素，依次比较其与堆顶的大小，若是比堆顶小，则堆顶弹出，再将新数加入堆中，直至数组结束，保证堆中的k个最小。
• step 3：最后将堆顶依次弹出即是最小的k个数。

---

图示理解：

---

代码解析：

---

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        //创建一个结果ArrayList res 用于存放最小的k个数
        ArrayList<Integer> res = new ArrayList<Integer>();
        //1.首先判断特殊情况，如果k等于0或者输入数组input的长度为0，则直接返回空的ArrayList。
        if (k == 0 || input.length == 0)
            return res;
        //2.创建一个大根堆 PriorityQueue q，用于存放当前最小的k个数。
        PriorityQueue<Integer> q = new PriorityQueue<>((o1, o2)->o2.compareTo(o1));
        //3.遍历数组input的前k个元素，将其添加到堆q中
        for (int i = 0; i < k; i++)
            q.offer(input[i]);
        //4.从第k+1个元素开始遍历数组input的剩余元素
        for (int i = k; i < input.length; i++) {
            //较小元素入堆
            //5.如果当前元素小于堆q中的最大值，即q.peek()，则将堆q中的最大值弹出，将当前元素加入堆q
            if (q.peek() > input[i]) {
                q.poll();
                q.offer(input[i]);
            }
        }//循环结束后，堆q中剩余的k个元素就是最小的k个数
        //6.最后，将堆q中的元素依次弹出并添加到结果ArrayList res中
        for (int i = 0; i < k; i++)
            res.add(q.poll());
        return res;
    }
}

总结