题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
tips:给定当前节点,求其在中序遍历的下一个节点,并且已知每个节点的父节点。
方法一,可以通过给定节点的父节点循环追溯至根节点,然后通过根节点进行中序遍历,找到给定节点的下一个节点。该方法需要遍历整棵树,复杂度较高。
方法二,我们只需遍历有限的几个分支即可找到下一个节点。中序遍历即按照左子树、根节点、右子树的顺序遍历。那么
①如果当前节点有右子树,下一个节点为右子树的最左边的叶子节点
②如果当前节点无右子树,向父节点追溯,若当前节点是父节点的左孩子,输出父节点。否则继续向上追溯,更新父节点为当前节点,继续判断当前节点与父节点的关系。边界判断,若追溯到根节点还未输出,继续循环,一般根节点的父节点为空,故无需单独判定边界,直接输出父节点即可。
代码如下:
# -*- 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):
# write code here
if not pNode:
return
if pNode.right:
pNode=pNode.right
while pNode.left:
pNode=pNode.left
return pNode
else:
parent=pNode.next
while parent and parent.left!=pNode:
parent,pNode=parent.next,parent
return parent