给定一个二叉树和其中的一个节点,请找出中序遍历顺序的下一个节点并且返回。书中的节点不仅包含左右子节点,同时包含指向父节点的指针。
class TreeLinkNode:
def __init__(self,x):
self.val = x
self.left = None
self.right =None
self.next =None
class Solution:
def GetNext(self,pNode):
# 1. 寻找右子树,如果存在就一直找到右子树的最左边,就是下一个节点
# 2. 没有右子树,就寻找它的父节点,一直找到它是父节点的左子树,打印父节点
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