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