描述
给定一个二叉树其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的next指针。下图为一棵有9个节点的二叉树。树中从父节点指向子节点的指针用实线表示,从子节点指向父节点的用虚线表示。
官方给的示例,比如输入:{8,6,10,5,7,9,11},8,中序遍历{5,6,7,8,9,10,11},根节点8的下一个节点就是9,应该返回{9,10,11},后台只打印子树的下一个节点,所以只会打印9。
那先按照输入的数字创建一棵树,然后中序遍历查找下一个结点。
来自https://blog.nowcoder.net/n/95afa7a2f4794af184bbe73e5ed61995?f=comment
class Solution:
def GetNext(self, pNode):
# write code here
# 中序遍历存储数组
res = []
cur = pNode
while cur.next:
cur = cur.next
self.inorder(cur, res)
for i in range(len(res)):
if pNode == res[i]:
# 判断 i 是不是最后一个结点,是则返回none,否则返回下一个数组结点
return None if i == len(res)-1 else res[i+1]
return None
# 递归遍历中序
def inorder(self, cur, res):
if cur:
self.inorder(cur.left, res)
res.append(cur)
self.inorder(cur.right, res)
时间复杂度O(N):N表示树的结点数量,递归遍历整个树结点,查找遍历数组,最差情况都是O(N)
空间复杂度O(N):存储二叉树中序遍历数组res占用空间O(N)
还有一种空间复杂度较小的解法:特性模拟
来自https://blog.nowcoder.net/n/27cc23d47d7247379bd8b8a4dd6a3607?f=comment
先判断目标结点是不是叶子结点?
①如果不是叶子结点,那么在中序遍历中,下一个结点就是当前结点右子树的最左叶子结点。
②如果是叶子结点,如果当前结点是父结点的左子树,那么就要往上寻找父结点。如果当前结点是父节点的右子结点,继续往上找,直到找到当前结点是父结点的左子结点。
class Solution:
def GetNext(self, pNode):
# write code here
if not pNode:
return pNode
# 1、如果目标结点不是叶子结点
while pNode.right:
p=pNode.right
# 则在中序遍历中,下一个结点为当前结点的右子树的最左叶子结点
if p.left:
p=p.left
return p
# 2、目标结点是叶子结点而不是根节点,则需要向上寻找父节点
while pNode.next:
p=pNode.next
# 是父节点的右子结点,则在中序遍历中,还要继续寻找,直到满足当前结点是父节点的左子结点
if p.left==pNode:
return p
pNode=p
时间复杂度 O(N):最坏的情况是遍历所有结点
空间复杂度 O(1):没有用到类似数组的类型数据