【剑指offer】二叉树的下一个结点(python)

分两种情况,一个是当前结点有右子树,一个是当前结点没有右子树。
有右子树:返回该右子树的最左结点,往左遍历即可
没有右子树:向上找父节点,该父节点的左孩子需等于当前结点,当前结点向上遍历。

class Solution:
    def GetNext(self, pNode):
        # write code here
        if pNode.right:
            p = pNode.right
            while p.left:
                p = p.left
            return p
        while pNode.next:
            if pNode.next.left == pNode:
                return pNode.next
            pNode = pNode.next
        return