(python version) 剑指offer 8.二叉树的下一个节点

题目叙述

給定一個二叉樹和其中的一個結點，請找出中序遍歷順序的下一個結點並且返回。注意，樹中的結點不僅包含左右子結點，同時包含指向父結點的指針。

Python代码

class TreeLinkNode:
    def __init__(self, x):
        self.val   = x
        self.left  = None
        self.right = None
        self.next  = None  # 指向父节点的指针

class Solution:
    def GetNext(self, pNode):
        """
        中序遍历顺序：左子-> 父节点-> 右子
        """

        if not pNode:
            return None

        # 如果节点有右子树，下一个节点 = 右子树的最左
        elif pNode.right:
            return self.getRightLeft(pNode.right)
        # 如果没有，下一个节点为父节点
        else:
            return self.getHead(pNode)

    def getHead(self, pNode):
        """
        找寻父节点
        1.终止条件：如果没有父节点指针，则为根节点
        2.如果该节点为父节点的左子节点，则返回父节点
        3.如果该节点为父节点的右子节点，向上遍历直到找到该节点为父节点的左子节点
        """

        # 父节点
        while pNode.next:
            # condition2
            if pNode == pNode.next.left:
                return pNode.next
            # condition3
            elif pNode == pNode.next.right:
                pNode = pNode.next
        # 根节点
        return None

    def getRightLeft(self, pNode):
        """
        找寻右子树的最左子节点遍历
        """
        while pNode.left:
            pNode = pNode.left
        return pNode