剑指Offer(Python多种思路实现):二叉树的下一个节点
面试8题:
题目:二叉树的下一个节点
题目描述:给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
解题思路一:
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
解题思路二:
def GetNext(self, pNode):
# write code here
if not pNode:
return None
# 有右子树,右子树中最左节点
if pNode.right:
pre = pNode.right
while pre.left:
pre = pre.left
return pre
while pNode.next:
parent = pNode.next
if parent.left == pNode:
return parent
pNode = parent
return None