【剑指offer刷题】--二叉树的下一个结点

题目描述：

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

 

解：我们都了解了中序遍历的规则是 左 根 右

# -*- 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

 在中序遍历中，找到下一个结点并返回，那我们并不需要考虑该结点是否有左子树。只要考虑该节点是否是右子树即可。

1. 当该结点有右子树时，比如B点，那结点B的下一个结点是它右子树的最左侧的孩子，即G点。
2. 当该结点没有右子树时，可以分为两种情况：

• 第一种情况，该节点是它父节点的左孩子结点，比如G点，它是父节点E的左孩子结点，那G点的下一个结点便是E点；
• 第二种情况，该结点是它父节点的右孩子结点，比如E点，它是父节点B的右孩子结点，很明显E的下一个结点是A，那我们要再往上找，B是A的左孩子结点