题解 | #第k轻的牛牛# java

第k轻的牛牛

https://www.nowcoder.com/practice/d3b31f055b1640d9b10de0a6f2b8e6f3

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 kthLighest (TreeNode root, int k) {
        // write code here
        Queue<TreeNode> queue = new LinkedList<>();
        List<Integer> res = new ArrayList<>();

        if (root != null) {
            queue.offer(root);
        }

        while (!queue.isEmpty()) {
            int size = queue.size();

            for (int i = 0; i < size; i++) {
                TreeNode cur = queue.poll();
                res.add(cur.val);

                if (cur.left != null) {
                    queue.offer(cur.left);
                }
                if (cur.right != null) {
                    queue.offer(cur.right);
                }
            }
        }

        Collections.sort(res);
        return res.get(k - 1);
    }
}

代码使用的编程语言是Java。

该题考察的知识点是二叉树的遍历和数据结构的操作。

这段代码实现了一个方法kthLighest,该方法接收一个树的根节点root和一个整数k作为参数,返回二叉树中第k大的节点的值。

代码的逻辑如下:

  1. 创建一个队列queue和一个列表res,用于存储遍历过程中的节点和节点值。
  2. 如果根节点root不为空,将其入队列。
  3. 使用广度优先搜索(BFS)遍历二叉树:获取当前队列中的节点个数size。循环遍历队列中的节点:弹出队列中的节点cur。将节点cur的值添加到列表res中。如果节点cur的左子节点不为空,将其加入队列。如果节点cur的右子节点不为空,将其加入队列。
  4. 对列表res进行排序。
  5. 返回列表res中第k-1个元素,即第k大的节点的值。
全部评论

相关推荐

SadnessAlex:跟三十五岁原则一样,人太多给这些***惯坏了
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务