题目描述:
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
解:我们都了解了中序遍历的规则是 左 根 右
# -*- 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):
if not pNode:
return
if not pNode.right:
return self.getHead(pNode)
return self.getRightHead(pNode.right)
def getHead(self,pNode):
while pNode.next:
if pNode.next.right==pNode:
pNode=pNode.next
elif pNode.next.left==pNode:
return pNode.next
return None
def getRightHead(self,pNode):
while pNode.left:
pNode=pNode.left
return pNode
在中序遍历中,找到下一个结点并返回,那我们并不需要考虑该结点是否有左子树。只要考虑该节点是否是右子树即可。
- 当该结点有右子树时,比如B点,那结点B的下一个结点是它右子树的最左侧的孩子,即G点。
- 当该结点没有右子树时,可以分为两种情况:
- 第一种情况,该节点是它父节点的左孩子结点,比如G点,它是父节点E的左孩子结点,那G点的下一个结点便是E点;
- 第二种情况,该结点是它父节点的右孩子结点,比如E点,它是父节点B的右孩子结点,很明显E的下一个结点是A,那我们要再往上找,B是A的左孩子结点