Python-二叉查找树的第k大的节点

解决策略：

遍历整棵二叉树然后对其进行排序，取第 k 大的节点

结合二叉查找树的本身特性：一个节点的左子树的所有节点值都小于当前节点值，右子树所有节点值都大于当前节点值

因此，对二叉查找树的中序遍历就是排序后的节点

代码（递归实现）：

# coding=utf-8

"""
question:
给定一颗二叉搜索(查找)树，找到其第 k 个节点
其中第 k 个节点指将所有节点排序(升序)后的第 k 个节点(从 0 计算)

例：
        4
   2       7
1    3   6    9
排序后为[1, 2, 3, 4, 6, 7, 9]
"""


# 节点类
class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left_child = None
        self.right_child = None

    # 返回以本节点为根的中序遍历
    def mid_traversal(self):
        ret = []

        # 遍历左子树
        if self.left_child:
            left_t = self.left_child.mid_traversal()
        else:
            left_t = []

        # 遍历右子树
        if self.right_child:
            right_t = self.right_child.mid_traversal()
        else:
            right_t = []

        ret.extend(left