题解 | #第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 背包问题 分治算法:归并排序、快速排序