题解 | #二叉树之寻找第k大#

二叉树之寻找第k大

https://www.nowcoder.com/practice/8e5f73fa3f1a407eb7d0b0d7a105805e

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @param k int整型 
     * @return int整型
     */
    public int kthLargest (TreeNode root, int k) {
        // write code here
        Queue<Integer> queue = new PriorityQueue<>();
        preOrder(root, queue, k);
        return queue.peek();
    }
    private void preOrder(TreeNode root, Queue<Integer> queue, int k) {
        if (root == null) return;
        queue.offer(root.val);
        if (queue.size() > k) queue.poll();
        preOrder(root.left, queue, k);
        preOrder(root.right, queue, k);
    }
}
  • 根据题意要找第k大的,可以想到用小顶堆来维护,这里使用Java自带的 PriorityQueue
  • 任意选择一种二叉树的遍历方式(前中后都可以),元素添加到 queue 后,如果queue的大小超过k,就移除堆顶元素
  • 最后小顶堆堆顶元素就是所求答案
  • #二叉树的遍历##优先队列的应用#
    线性表基础 文章被收录于专栏

    链表、递归、栈

    全部评论

    相关推荐

    不愿透露姓名的神秘牛友
    07-07 11:30
    仁者伍敌:kpi都懒得刷了属于是
    点赞 评论 收藏
    分享
    05-11 11:48
    河南大学 Java
    程序员牛肉:我是26届的双非。目前有两段实习经历,大三上去的美团,现在来字节了,做的是国际电商的营销业务。希望我的经历对你有用。 1.好好做你的CSDN,最好是直接转微信公众号。因为这本质上是一个很好的展示自己技术热情的证据。我当时也是烂大街项目(网盘+鱼皮的一个项目)+零实习去面试美团,但是当时我的CSDN阅读量超百万,微信公众号阅读量40万。面试的时候面试官就告诉我说觉得我对技术挺有激情的。可以看看我主页的美团面试面经。 因此花点时间好好做这个知识分享,最好是单拉出来搞一个板块。各大公司都极其看中知识落地的能力。 可以看看我的简历对于博客的描述。这个帖子里面有:https://www.nowcoder.com/discuss/745348200596324352?sourceSSR=users 2.实习经历有一些东西删除了,目前看来你的产出其实很少。有些内容其实很扯淡,最好不要保留。有一些点你可能觉得很牛逼,但是面试官眼里是减分的。 你还能负责数据库表的设计?这个公司得垃圾成啥样子,才能让一个实习生介入数据库表的设计,不要写这种东西。 一个公司的财务审批系统应该是很稳定的吧?为什么你去了才有RBAC权限设计?那这个公司之前是怎么处理权限分离的?这些东西看着都有点扯淡了。 还有就是使用Redis实现轻量级的消息队列?那为什么这一块不使用专业的MQ呢?为什么要使用redis,这些一定要清楚, 就目前看来,其实你的这个实习技术还不错。不要太焦虑。就是有一些内容有点虚了。可以考虑从PR中再投一点产出
    投递美团等公司8个岗位
    点赞 评论 收藏
    分享
    评论
    点赞
    收藏
    分享

    创作者周榜

    更多
    牛客网
    牛客网在线编程
    牛客网题解
    牛客企业服务