剑指offer—二叉树的下一个结点（Python版）

题目描述：

给定一个二叉树和其中一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

解题思路：

二叉树的中序遍历顺序：LvR（左子树、根节点、右子树）

结合上图，我们发现可分为两种情况：
1、有右子树的，那么下个节点就是右子树最左边的点；
2、没有右子树的，也可以分成两类：
a) 是父节点左孩子，那么父节点就是下一个节点 ；
b) 是父节点的右孩子，找他的父节点的父节点的父节点…直到当前结点是其父节点的左孩子位置。
如果没有，那么它就是尾节点。

代码实现：

# -*- coding:utf-8 -*-
# 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 pNode
        if pNode.right:
            tmpNode = pNode.right
            while tmpNode.left:
                tmpNode = tmpNode.left
            return tmpNode
        else:
            tmpNode = pNode
            while tmpNode.next:
                if tmpNode.next.left == tmpNode:
                    return tmpNode.next
                tmpNode = tmpNode.next
            return None