JZ62 二叉搜索树的第k个结点

原创于 2021-09-02 17:44:25 发布 · 160 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #算法 #python

剑指offer 专栏收录该内容

67 篇文章

订阅专栏

本文介绍如何在二叉搜索树中找到第k小的TreeNode节点，通过中序遍历策略实现，并讨论了边界条件和返回TreeNode节点的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述

给定一棵二叉搜索树，请找出其中的第k小的TreeNode结点。

有点像JZ14 链表中倒数最后k个结点，第k小的结点，那就是按照结点值从小到大的顺序排列后，第k位的结点。按照树的遍历，中序遍历满足这个顺序。

题解中有一个很清晰的思路：

其实就是中序遍历，然后返回第k个值，但前提是要对k进行判定，如果k小于0或者k超出树结点的范围就返回None。

另外需要注意的是返回的是结点不是结点的值，所以在中序遍历中都是append(root)，而不是root.val。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回对应节点TreeNode
    def KthNode(self, pRoot, k):
        self.res=[]
        self.dfs(pRoot)
        return self.res[k-1] if 0<k<=len(self.res) else None
    def dfs(self,root):
        if not root:return
        self.dfs(root.left)
        self.res.append(root)
        self.dfs(root.right)