题解 | #第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);
                }
            }
        }
        res.sort(null);
        return res.get(k - 1);
    }
}

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

该题考察的知识点是二叉树的层序遍历和查找第k小的节点值。

代码的解释如下:

  1. 首先定义了一个TreeNode结构体,用于表示二叉树的节点。包括节点的值(val)、左子节点(left)和右子节点(right)。
  2. 接着定义了一个Solution类,包含了一个kthLighest方法,用于查找给定二叉树中第k小的节点值。
  3. kthLighest方法中,使用队列que来进行层序遍历。将根节点加入队列,然后循环直到队列为空。
  4. 每次循环首先获取当前层的节点数size,然后在循环中取出当前层所有节点。
  5. 将当前节点的值加入结果列表res中,并将当前节点的左子节点和右子节点加入队列。
  6. 遍历完所有层后,对结果列表进行排序,然后返回第k个最小的节点值。

总结起来,该段代码实现了使用队列进行层序遍历,将二叉树所有节点值存储在列表中,并通过对列表进行排序获得第k小的节点值。

全部评论

相关推荐

点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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