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

第k轻的牛牛

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

在代码中,kthLighest 方法接受一个二叉搜索树的根节点 root 和一个整数 k 作为输入,并返回第 k 个最小元素的值。

方法内部使用了中序遍历二叉搜索树,并将遍历得到的节点值存储在一个列表 list 中。由于中序遍历得到的节点值是升序排列的,所以第 k 个最小元素对应列表中的索引为 k - 1

首先,我们创建一个空列表 list 来存储节点值。

然后,调用私有方法 inorderTraversal 对二叉搜索树进行中序遍历。在中序遍历中,首先递归地遍历左子树,然后将当前节点值添加到列表中,最后递归地遍历右子树。

最后,返回列表 list 中第 k - 1 个元素的值。

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) {
        List<Integer> list = new ArrayList<>();
        inorderTraversal(root, list);
        return list.get(k - 1);
    }

    private void inorderTraversal(TreeNode node, List<Integer> list) {
        if (node == null) {
            return;
        }

        inorderTraversal(node.left, list); // 先遍历左子树
        list.add(node.val); // 将当前节点值添加到列表中
        inorderTraversal(node.right, list); // 遍历右子树
    }
}

算法题刷刷刷 文章被收录于专栏

数组、链表、栈、队列、堆、树、图等。 查找和排序:二分查找、线性查找、快速排序、归并排序、堆排序等。 动态规划:背包问题、最长公共子序列、最短路径 贪心算法:活动选择、霍夫曼编码 图:深度优先搜索、广度优先搜索、拓扑排序、最短路径算法(如 Dijkstra、Floyd-Warshall) 字符串操作:KMP 算法、正则表达式匹配 回溯算法:八皇后问题、0-1 背包问题 分治算法:归并排序、快速排序

全部评论

相关推荐

Gaynes:查看图片
点赞 评论 收藏
分享
06-15 18:44
黄淮学院 Java
Lynn012:如果是居民楼还是算了吧,看着有点野呢
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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