剑指offer 二叉搜索树的第k个结点

sparksnail

于 2018-06-22 21:09:40 发布

阅读量132

点赞数

分类专栏：剑指offer

本文链接：https://blog.csdn.net/y12345678904/article/details/80778694

版权

剑指offer 专栏收录该内容

104 篇文章 0 订阅

订阅专栏

本文介绍如何在二叉搜索树中找到第k小的节点。通过二叉树的中序遍历，可以按节点数值大小顺序轻松找到第k小的节点。在给定的示例树中，第3个最小节点的值是4。

摘要由CSDN通过智能技术生成

题目

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

思路

二叉树中序遍历

代码

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回对应节点TreeNode
    def __init__(self):
        self.count = 0
        self.res_node = None
    def find(self, pRoot, k):
        if pRoot:
            self.find(pRoot.left, k)
            self.count += 1
            if self.count == k:
                self.res_node = pRoot
            else:
                self.find(pRoot.right, k)
    def KthNode(self, pRoot, k):
        # write code here
        self.find(pRoot, k)
        return self.res_node