v1
//分析二叉树的下一个节点,一共有以下情况:
//1.二叉树为空,则返回空;
//2.节点右孩子存在,则设置一个指针从该节点的右孩子出发,一直沿着指向左子结点的指针找到的叶子节点即为下一个节点;
//3.节点不是根节点。如果该节点是其父节点的左孩子,则返回父节点;否则继续向上遍历其父节点的父节点,重复之前的判断,返回结果。
class Solution:
def GetNext(self, pNode):
# write code here
if not pNode:
return pNode
# 考虑结点是根节点,存在右子树的情况,那么返回的值就在右子树之间。
# 在右子树一直往左遍历,当左节点不存在的时候,就返回当前结点。(中序遍历)
if pNode.right:
Node = pNode.right
while Node.left:
Node = Node.left
return Node
# 考虑结点是某个结点的子节点的情况,这种情况需要往上一直搜索,直到是左子树
par = pNode.next
while par:
if par.left == pNode:
return par
else:
pNode = par
par = par.next
par = par.next
return None