AC Wing：19.二叉树的下一个节点（python版）

最新推荐文章于 2020-06-16 18:52:00 发布

原创最新推荐文章于 2020-06-16 18:52:00 发布 · 565 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #数据结构 #算法

算法专栏收录该内容

34 篇文章

订阅专栏

本文探讨了在给定二叉树的一个节点时，如何找到中序遍历序列中的下一个节点。通过分析三种情况，即有右子树、无右子树但为左孩子、无右子树且为右孩子，提供了详细的解决方案和Python代码实现。

题目描述

给定一棵二叉树的其中一个节点，请找出中序遍历序列的下一个节点。

注意：

如果给定的节点是中序遍历序列的最后一个，则返回空节点;
二叉树一定不为空，且给定的节点一定不是空节点；

样例：
假定二叉树是：[2, 1, 3, null, null, null, null]，给出的是值等于2的节点。
则应返回值等于3的节点。

思路

分三种情况讨论：

给定节点有右子树，则其下一个节点为该节点右子树的最左孩子节点
给定节点无右子树，且给定节点为左孩子节点，则其下一个节点为该节点的父亲节点
给定节点无右子树，且给定节点为右孩子节点，则沿着给定节点的父亲节点一直往上找，最后的父亲节点为该节点的下一个节点

代码实现（Python）

# Definition for a binary tree node.
class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
        self.father = None
class Solution(object):
    def inorderSuccessor(self, p):
        """
        :type p: TreeNode
        :rtype: TreeNode
        """
        if p.right is not None:     # 1、有右子树
            p = p.right             # p指向p的右子树
            while p.left is not None:
                p = p.left          # 找右树的最左儿子
            return p
        else:   # 无右子树
            while p.father is not None:
                if p == p.father.left:   # 2、无右子树且为左儿子
                    return p.father
                else:                    # 3、无右子树且为右儿子
                    p = p.father
            return None