题目
给定一个二叉树和其中一个节点,找出该节点在中序遍历序列中的下一个节点。
注意:
树中的节点除了有两个分别指向左右子节点的指针,还有一个指向父节点的指针。
如果给定的节点是中序遍历序列的最后一个,则返回空节点。
二叉树一定不为空,且给定的节点一定不是空节点。
分情况讨论即可,如下图所示:
思路
- 如果当前节点有右孩子,则右子树中最左侧的节点就是当前节点的后继。 比如F的后继不是E而是H。寻找过程和中序遍历找最左子节点的过程是一致的。
- 如果当前节点没有右孩子,则需要沿着father域一直向上找。比如A的后继是其父节点C,但D的后继是父节点的父节点F。
- 找到第一个是其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即后继节点