【剑指offer】JZ8 二叉树的下一个结点

1 问题

给定一个二叉树其中的一个结点（不一定是根节点），请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的next指针。下图为一棵有9个节点的二叉树。树中从父节点指向子节点的指针用实线表示，从子节点指向父节点的用虚线表示

示例:
输入:{8,6,10,5,7,9,11},8
返回:9
解析:这个组装传入的子树根节点，其实就是整颗树，中序遍历{5,6,7,8,9,10,11}，根节点8的下一个节点就是9，应该返回{9,10,11}，后台只打印子树的下一个节点，所以只会打印9，如下图，其实都有指向左右孩子的指针，还有指向父节点的指针，下图没有画出来

要求：空间复杂度 O(1) ，时间复杂度O(n)

输入描述：
输入分为2段，第一段是整体的二叉树，第二段是给定二叉树节点的值，后台会将这2个参数组装为一个二叉树局部的
子树传入到函数GetNext里面，用户得到的输入只有一个子树根节点

返回值描述：
返回传入的子树根节点的下一个节点，后台会打印输出这个节点

示例1
输入：{8,6,10,5,7,9,11},8
返回值：9

示例2
输入：{8,6,10,5,7,9,11},6
返回值：7

示例3
输入：{1,2,#,#,3,#,4},4
返回值：1

示例4
输入：{5},5
返回值：“null”
说明：不存在，后台打印"null"

2 答案

自己写的不对，中序遍历递归算法，没理解输入是怎么回事，要先找到根节点，根节点是最上面的，这题有个next指针，和正常的二叉树不太一样

class Solution:
    def GetNext(self, pNode):
        def dfs(root, target):
            if not root:
                return None
            return dfs(root.left, target)
            if root.val == target:
                return root.right.val
            return dfs(root.right, target)

        return dfs(pNode[0], pNode[1])

官方解

1. 中序遍历递归，注意要先找到根节点，再得到中序遍历，再遍历得到的列表

我们首先要根据给定输入中的结点指针向父级进行迭代，直到找到该树的根节点；然后根据根节点进行中序遍历，当遍历到和给定树节点相同的节点时，下一个节点就是我们的目标返回节点

• step 1：首先先根据当前给出的结点找到根节点
• step 2：然后根节点调用中序遍历
• step 3：将中序遍历结果存储下来
• step 4：最终拿当前结点匹配是否有符合要求的下一个结点

class Solution:
    nodes = []

    def GetNext(self, pNode):
        # 查找根节点
        root = pNode
        while root.next:
            root = root.next

        # 中序遍历打造nodes
        self.InOrder(root)

        # 匹配节点
        for i in range(len(self.nodes) - 1):
            if pNode == self.nodes[i]:
                return self.nodes[i + 1]
        return None

    # 中序遍历
    def InOrder(self, root):
        if root == None:
            return
        self.InOrder(root.left)
        self.nodes.append(root)  # 为什么不能写root.val，因为root已经是整数了？
        self.InOrder(root.right)

2. 分类直接寻找

直接寻找分为三种情况

• 如果给出的结点有右子节点，则最终要返回的下一个结点即右子树的最左下的结点

• 如果给出的结点无右子节点，且当前结点是其父节点的左子节点，则返回其父节点

• 如果给出的结点无右子节点，且当前结点是其父节点的右子节点，则先要沿着左上方父节点爬树，一直爬到当前结点是其父节点的左子节点为止，返回的就是这个父节点；或者没有满足上述情况的则返回为NULL（和情况二类似）

• step 1：判断该节点是否符合思路中第一点，则一直找到右子树的左下节点为返回值

• step 2：判断该节点是否符合思路中第二点，则返回当前节点的父亲节点

• step 3：判断该节点是否符合思路中第三点，则迭代向上找父节点，直到迭代的当前节点是父节点的左孩子节点为止，返回该父节点；如果不满足上述情况返回NULL

class Solution:
    def GetNext(self, pNode):
        # 情况一
        if pNode.right:
            rchild = pNode.right
            # 一直找到右子树的最左下的结点为返回值
            while rchild.left:
                rchild = rchild.left
            return rchild
        # 情况二
        # 如果无右子树且当前结点是其父节点的左子结点
        if pNode.next and pNode.next.left == pNode:
            return pNode.next

        # 情况三
        if pNode.next:
            ppar = pNode.next
            # 沿着左上一直爬树，爬到当前结点是其父节点的左自己结点为止
            while ppar.next and ppar.next.right == ppar:
                ppar = ppar.next
            return ppar.next

        return None

https://www.nowcoder.com/share/jump/9318638301698325539037