题目:
剑指 Offer 54. 二叉搜索树的第 k 大节点。
给定一棵二叉搜索树,请找出其中第 k 大的节点的值。
链接:https://leetcode-cn.com/leetbook/read/illustrate-lcof/xspy85/
示例:
限制: 1 ≤ k ≤ 二叉搜索树元素个数
思路:
由于二叉搜索树中值的大小为右节点>根节点>左节点,于是按照“右中左”的顺序查找,每次将k-1,直到k等于0时,说明找到了第k大的节点。
代码:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def kthLargest(self, root: TreeNode, k: int) -> int:
def dfs(root):
if not root: return
dfs(root.right)
if self.k == 0: return
self.k -= 1
if self.k == 0:
self.res = root.val
dfs(root.left)
self.k = k
dfs(root)
return self.res
时间复杂度O(n),空间复杂度O(1).