//分析二叉树的下一个节点,一共有以下情况:
//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
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)