二叉树的下一个节点

题目：二叉树的下一个节点

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

# -*- coding:utf-8 -*-
# class TreeLinkNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#         self.next = None

在二叉树的初始化时，二叉树x可以轻松得到父节点、左子树、右子树、根节点值。中序遍历的结果：左子树、根节点、右子树。寻找该节点的中序遍历顺序的下一个节点，根据该节点的位置，可分为不同情况：

（1）该节点有左子树，说明左边已经遍历过，该节点作为根节点，下一步应该填右子树，如果该节点有右子树，下一个节点出现在右子树（出现在右子树的最早被遍历的节点——最左节点），进入到该右子树，寻找最左节点。

        if pNode.right!=None:
            pNode=pNode.right
            while pNode.left!=None:
                pNode=pNode.left
            return pNode

（2）如果该节点没有右子树，下一个节点可能出现把该节点作为左子树的父节点，也就是说，如果一个节点没有右子树，并且它还是它父节点的左子节点，则下一个节点就是其父节点

               return pNode.next

（3）如果一个节点没有右子树，并且它还是它父节点的右子节点，需要沿着指向父节点的指针向上遍历，需要找到一个祖节点为其父节点的左子节点，则下一个节点就是祖节点的父节点。

if pNode.next!=None and pNode.next.right==pNode:
    while pNode.next!=None and pNode.next.left!=pNode:
        pNode=pNode.next
    return pNode.next

最终程序：

class Solution:
    def GetNext(self, pNode):
        # write code here
        if not pNode:
            return
        #如果该节点有右子树，那么下一个节点就是它右子树中的最左节点
        elif pNode.right!=None:
            pNode=pNode.right
            while pNode.left!=None:
                pNode=pNode.left
            return pNode
        
        #如果一个节点没有右子树，，并且它还是它父节点的右子节点
        elif pNode.next!=None and pNode.next.right==pNode:
            while pNode.next!=None and pNode.next.left!=pNode:
                pNode=pNode.next
            return pNode.next
        #如果一个节点是最后一个节点
        else:
            return pNode.next