230. 二叉搜索树中第K小的元素


230. 二叉搜索树中第K小的元素

题目描述

在这里插入图片描述

解题思路

  1. 二叉搜索树的中序遍历是有序(升序)的,利用这个性质。
  • 二叉搜索树性质:
  1. 左子树节点值均比根节点值小,右子树节点值均比根节点值大。这是解题思路的来源。
  2. 左右子树均保持性质1。

代码实现

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    # ptr = 0
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        """ 
        二叉搜索树的中序遍历是有序(升序)的,利用这个性质。
        内部函数使用到k和返回结果,需要全局变量,因此self.k = k; self.res = None

        """
        self.k = k
        self.res = None
        def inorder(root: TreeNode) -> List[int]:
            if root is None:
                return
            inorder(root.left)
            self.k -= 1
            if self.k == 0:
                self.res = root.val
            inorder(root.right)
        inorder(root)
        return self.res


    def kthSmallest1(self, root: TreeNode, k: int) -> int:
        # 二叉搜索树的中序遍历是有序(升序)的,利用这个性质即可
        def inorder(root: TreeNode) -> List[int]:
            if root is None:
                return []
            return inorder(root.left) + [root.val] + inorder(root.right)

        return inorder(root)[k - 1]

上面算法最坏情况每个节点遍历一遍,时间复杂度O(N)。
一个高效的优化思路:在每个BST树的节点中维护排名,这样可以直接比较排名,时间复杂度O(logN)。

参考文献

  1. 手把手刷二叉搜索树(第一期)
  2. 官方题解
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页