二叉搜索树的第k个节点

62.二叉搜索树的第k个节点

题目描述

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

解题思路

二叉搜索树已经是一棵排序好的树，中序遍历即可得到一组排序好的节点。然后根据题意获取第k个节点

Python代码如下

# -*- 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):
        # write code here
        self.res = []
        self.search(pRoot)
        if k>0 and k<=len(self.res):
            return self.res[k-1]
        else:
            return None


    def search(self,root):
        if not root:
            return None
        self.search(root.left)
        self.res.append(root)
        self.search(root.right)