牛客-剑指offer系列题解:二叉树的下一个结点

最新推荐文章于 2022-05-30 11:46:42 发布

剑圣土豆

最新推荐文章于 2022-05-30 11:46:42 发布

阅读量239

点赞数

分类专栏：牛客-剑指offer系列题解

本文链接：https://blog.youkuaiyun.com/weixin_42042056/article/details/105924714

版权

牛客-剑指offer系列题解专栏收录该内容

66 篇文章

订阅专栏

本文详细解析了如何在二叉树中寻找中序遍历的下一个节点，介绍了当右子树存在时找最左节点，以及右子树不存在时向上查找父节点的策略，适用于Python编程环境。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

记录刷题的过程。牛客和力扣中都有相关题目，这里以牛客的题目描述为主。该系列默认采用python语言。
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):
        # write code here
        #如果右子树存在，则找右子树的最左边的节点，则返回该节点
        #如果右子树不存在，则找其父节点，一直找到他是根节点的左子树，则返回根节点
        if pNode.right:
            ptemp = pNode.right
            while ptemp.left:
                    ptemp = ptemp.left
            return ptemp
        else:
            ptemp = pNode
            while ptemp.next:
                if ptemp.next.left == ptemp:
                    return ptemp.next
                ptemp = ptemp.next
            return None