题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:1.二叉树为空,返回空
2.# 节点右孩子存在,则设置一个指针从该节点的右孩子出发,一直沿着指向左孩子的指针找到的没有左孩子的节点(不一定为叶子节点,也可以有右孩子)即为下一个节点
3. # 节点不是根节点,(因为上面讨论了节点有孩子存在,这里节点右孩子都不存在)若该节点为其父节点的左孩子,则返回父节点,否则继续向上遍历,直到找到其最近父节点,如果其最近父节点是其最近父节点的父节点的左孩子,返回最近父节点的父节点。重复之前的判断
# -*- 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 None
if pNode.right:
pNode = pNode.right
while pNode.left:
pNode = pNode.left
return pNode
while pNode.next:
proot = pNode.next
if proot.left == pNode:
return proot
pNode = pNode.next **# 注意这里不是pNode = pNode.next.next,**例:只有{4,2,5,1,3}在节点为5时,下一个节点本为1,因为pNode = pNode.next.next为1,而proot为None 造成错误
return None