剑指offer：二叉搜索树的第k个节点（Python）

题目描述

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

解题思路

题中给的树的形状如下：

为什么 [ 按结点数值大小顺序第三个结点的值为4 ] ？
题中没有直接传达的一个信息是：给定一颗平衡二叉树，按照中序遍历的顺序，获得其第k大的节点。
基于以上信息开始求解，会发现这几乎就是一道中序遍历的题，不同之处仅仅在于：中序遍历返回二叉树中序遍历后所有节点的值的列表，而这道题返回中序遍历过程中填入列表的第k个节点。我们当然可以先对平衡二叉树进行中序遍历，在返回的列表中找到第k个节点；但很显然还有更直接的方式，即中序遍历到需要的位置后立即返回，而不需要遍历完整棵二叉树。同样的，代码和二叉树的中序遍历很像，可以对比来看：

中序遍历的代码

self.array = []    
def midOrder(self, root):
    if not root:
        return self.array
    self.midOrder(root.left)
    self.array.append(root.val)
    self.midOrder(root.right)

本题的代码

count = 0
def KthNode(self, pRoot, k):
    if not pRoot:
        return None
    node = self.KthNode(pRoot.left, k)
    if node:
        return node
    self.count += 1
    if self.count == k:
        return pRoot
    node = self.KthNode(pRoot.right, k)
    if node:
        return node

对比来看，思路不要太清晰：
中序遍历一共三步：
1. 递归左子树
2. 存储根节点的值
3. 递归右子树

本题也需要三步：
1. 递归左子树，并判断有无返回节点。如果有，停止递归，返回所要返回的节点。
2. 当左子树没有返回节点时，判断当前的根节点是不是第k个遍历到的值（即第k大）。如果是，则返回该节点，停止递归。
3. 当左子树和根节点都没有返回节点时，递归右子树，并判断有无返回节点。如果有，停止递归，返回所要返回的节点。