题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路
分情况依次判断。
1. 若给定结点为空节点,则返回空
2. 若给定结点有右子树,则将结点指向右子树的根节点,然后找到其最左边的叶节点并返回
3. 若给定结点没有父节点,则返回空
4. 若给定结点有父节点,若给定结点是其父节点的左孩子,则返回其父节点;否则循环地找父节点的父节点直到当前结点是其父节点的左孩子,则返回其父节点。
代码
# -*- 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 pNode is None:
return None
while pNode.right is not None:
pNode = pNode.right
while pNode.left is not None:
pNode = pNode.left
return pNode
while pNode.next is not None:
if pNode.next.left is pNode:
return pNode.next
return None