AcWing 19. 二叉树的下一个节点

题目描述

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

注意：

• 如果给定的节点是中序遍历序列的最后一个，则返回空节点;
• 二叉树一定不为空，且给定的节点一定不是空节点；

样例

假定二叉树是：[2, 1, 3, null, null, null, null]， 给出的是值等于2的节点。

则应返回值等于3的节点。

解释：该二叉树的结构如下，2的后继节点是3。
  2
 / \
1   3

 算法

若当前节点有右子树，则右子树中最左侧的节点就是当前节点的后继；
若当前节点没有右子树，则需要沿着父级一直向上找，找到第一个是其父级左儿子的节点，该节点的父节点就是当前节点的后继。（可以这么理解，如果当前节点没有右子树，说明以当前节点为根节点的子树已经遍历完毕，那么向上找当前节点的父节点，如果当前节点是父节点的右子树，说明以父节点为根节点的子树也已遍历完毕，以此类推，直到找到节点是其父节点的左孩子的节点，此时找到的节点的父级节点就是我们要找的后继。）
时间复杂度分析
遍历的节点数永远不会大于树的高度。所以时间复杂度是 O(h)

Python3 代码

### Python3 代码
# 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
        """
        if q:
            root = q.right
            # 判断是否有右子树
            if root:
                # 如果有，右子树的最左侧
                while root and root.left:
                    root = root.left
                return root
            else:
                # 如果没有，判断是否是其父节点的左孩子
                while q.father and q != q.father.left:
                    q = q.father
                return q.father
        return None