二叉树的下一个结点

描述

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

解释

二叉树的下一个结点就是中序遍历的所给结点的下一个节点，此题由于还包含指向父节点的指针，所以可以将时间复杂度由O(n)降为O(h)

情况一：所给结点有右子树，返回右子树的最左结点
情况二：所给结点没有右子树，向上找，直到第一次找到当前节点是它的父节点的左孩子的节点，此节点的父节点就是所给结点的中序遍历的下一个节点，即所求。

代码

# -*- 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):
        # write code here
        # 有右子树，则中序的下一个是右子树的最左结点
        if pNode.right:
            p=pNode.right
            while p.left:
                p=p.left
            return p
        #无右子树，则找第一个当前节点是父节点左孩子的节点，返回该节点的父节点即所求
        while pNode.next and pNode.next.right == pNode:
            pNode = pNode.next
        return pNode.next #到了根节点仍没找到，则返回空