【剑指offer】二叉树的下一个结点(模拟)

题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。

模拟题,考虑几种情况:
1.首先最容易解决的情况当然是,当当前结点存在右子树时,直接返回右边结点
当有边界点存在左子树时,遍历到右子树的左子树叶,即下一个结点。

2.其次情况,当前结点不存在右子树时,只能向上回溯,而向上回溯时又分为两种情况:
当是从左子树回溯时,可以直接返回其父节点
当是从右子树回溯时,其父节点并不是下一个结点,必须继续向上回溯,回溯的过程中,若遇到相同情况,继续回溯,若遇到某个父节点存在右子树,执行第一种情况的解决方法。

# -*- 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.right!=None:
            pNode=pNode.right
            while pNode.left!=None:
                pNode=pNode.left
            return pNode
        while pNode.next!=None:
            fa=pNode.next
            if fa.left==pNode:
                return fa
            if fa.right!=None and fa.right!=pNode:
                pNode=fa.right
                while pNode.left!=None:
                    pNode=pNode.left
                return pNode
            pNode=fa
        return None
原文地址:https://www.cnblogs.com/kuronekonano/p/11794296.html