给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:中序遍历 左结点→根结点→右结点
- 首先观察有无右节点,如果有右节点,则先指向右节点再指向最后一个左节点
- 若没有右节点,则向前找根节点,找到第一个当前结点是父节点左孩子的节点
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 #到了根节点仍没找到,则返回空