[剑指offer] 8.二叉树的下一个节点

题目

给定一个二叉树和其中一个节点，找出该节点在中序遍历序列中的下一个节点。

注意：
树中的节点除了有两个分别指向左右子节点的指针，还有一个指向父节点的指针。
如果给定的节点是中序遍历序列的最后一个，则返回空节点。
二叉树一定不为空，且给定的节点一定不是空节点。

分情况讨论即可，如下图所示：

思路

1. 如果当前节点有右孩子，则右子树中最左侧的节点就是当前节点的后继。 比如F的后继不是E而是H。寻找过程和中序遍历找最左子节点的过程是一致的。
2. 如果当前节点没有右孩子，则需要沿着father域一直向上找。比如A的后继是其父节点C，但D的后继是父节点的父节点F。
3. 找到第一个是其father左孩子的节点，该节点的father就是当前节点的后继。比如当前节点是D，则D的father域中第一个满足是其father左孩子的节点是C，则C的父节点F即为D的后继。

【因为中序是左孩子–根--右孩子，从左孩子回溯到的父节点才有可能是后继节点，从右孩子回溯到的父节点一定是前驱节点】

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#         self.father = None       ####### 这里和之前不同
class Solution(object):
    def inorderSuccessor(self, q):
        """
        :type q: TreeNode
        :rtype: TreeNode
        """
        # 如果 q有右孩子节点，则寻找其右子树中的最左子节点
        if q.right:
            q = q.right   # 后继不一定是其右孩子节点！需要找！
            while q.left:
                q = q.left
            return q
        # 如果 q 无右孩子节点，则向上看父节点
        # 如果 q 是父节点的右孩子节点，则继续向上寻找
        while q.father and q == q.father.right:
            q = q.father
        # 跳出循环的几种情况：
        # [1] q是根节点（q.father==None），[2] q是其父节点的左孩子
        return q.father
        # [1] 根节点了仍然没跳出说明该节点在右子树的最右节点，是整个二叉树的结尾，没有后继节点，因此返回q.father即None
        # [2] q是其父节点的左孩子，返回父节点q.father即后继节点