老卫带你学---剑指offer刷题系列(57.二叉树的下一个结点)

最新推荐文章于 2020-07-14 17:29:16 发布

老卫带你学

最新推荐文章于 2020-07-14 17:29:16 发布

阅读量112

点赞数

分类专栏：剑指offer python

本文链接：https://blog.csdn.net/yixieling4397/article/details/105114915

版权

python 同时被 2 个专栏收录

158 篇文章

订阅专栏

剑指offer

63 篇文章

订阅专栏

本文详细解析了如何在二叉树中找到给定节点中序遍历的下一个节点，通过区分有无右子树及是否为父节点左孩子的不同情况，提供了一种有效的算法实现。

摘要由CSDN通过智能技术生成

57.二叉树的下一个结点

问题：

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

解决：

思想：

在这里插入图片描述
结合图，我们可发现分成两大类：1、有右子树的，那么下个结点就是右子树最左边的点；（eg：D，B，E，A，C，G） 2、没有右子树的，也可以分成两类，a)是父节点左孩子（eg：N，I，L），那么父节点就是下一个节点； b)是父节点的右孩子（eg：H，J，K，M）找他的父节点的父节点的父节点…直到当前结点是其父节点的左孩子位置。如果没有eg：M，那么他就是尾节点。

python代码：

# -*- 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 pNode
        if(pNode.right):
            pr=pNode.right
            while(pr.left):
                pr=pr.left
            return pr
        while(pNode.next):
            tmp=pNode.next
            if(tmp.left==pNode):
                return tmp
            pNode=tmp