剑指offer--面试题63：二叉搜索树的第k个结点

最新推荐文章于 2024-05-27 21:18:06 发布

qingyujean

最新推荐文章于 2024-05-27 21:18:06 发布

阅读量293

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer 文章标签：剑指offer

本文链接：https://blog.youkuaiyun.com/qingyujean/article/details/52691752

剑指offer 专栏收录该内容

63 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中寻找第K小结点的方法。通过中序遍历的方式，可以有效地找到指定排名的结点值。此方法适用于需要频繁查找特定排名结点的应用场景。

题目描述

给定一颗二叉搜索树，请找出其中的第k小的结点。例如， 5 / \ 3 7 /\ /\ 2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。

python实现：

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回对应节点TreeNode
    #法2：如果需要KthNode（）的调用很频繁，则参考：https://leetcode.com/problems/kth-smallest-element-in-a-bst/
    result = None
    cnt = 0
    def KthNode(self, pRoot, k):
        # write code here
        if k<=0 or pRoot is None:
            return None
        self.cnt = k
        self.inOrder(pRoot)
        return self.result
         
         
    def inOrder(self, root):
        if self.result is not None:
            return
        if root:
            self.inOrder(root.left)
            self.cnt -= 1
            #print root.val, self.cnt
            if self.cnt==0:
                #self.result = root.val
                self.result = root
                return
            self.inOrder(root.right)