Links

54.二叉搜索树的第k大节点

一、 题目

给定一棵二叉搜索树,请找出其中第 k 大的节点的值。
示例 1:
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出: 4
示例 2:
输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 4 限制:
1 ≤ k ≤ 二叉搜索树元素个数
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二、 分析

根据二叉搜索树的特性,我们通过中序遍历得到有序的数组就可以轻松获得结果
func kthLargest(_ root: TreeNode?, _ k: Int) -> Int {
var inorderArr: [Int] = []
func inorder(_ node: TreeNode?) {
guard let n = node else {
return
}
inorder(n.right)
inorderArr.append(n.val)
inorder(n.left)
}
inorder(root)
return inorderArr[k - 1]
}