JZ57 二叉树的下一个结点

描述

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

官方给的示例，比如输入:{8,6,10,5,7,9,11},8，中序遍历{5,6,7,8,9,10,11}，根节点8的下一个节点就是9，应该返回{9,10,11}，后台只打印子树的下一个节点，所以只会打印9。

那先按照输入的数字创建一棵树，然后中序遍历查找下一个结点。

来自https://blog.nowcoder.net/n/95afa7a2f4794af184bbe73e5ed61995?f=comment

class Solution:
    def GetNext(self, pNode):
        # write code here
        # 中序遍历存储数组
        res = []
        cur = pNode
        while cur.next:
            cur = cur.next
        self.inorder(cur, res)
        
        for i in range(len(res)):
            if pNode == res[i]:
                # 判断 i 是不是最后一个结点，是则返回none，否则返回下一个数组结点
                return None if i == len(res)-1 else res[i+1]
        return None
    # 递归遍历中序
    def inorder(self, cur, res):
        if cur:
            self.inorder(cur.left, res)
            res.append(cur)
            self.inorder(cur.right, res)

时间复杂度O(N)：N表示树的结点数量，递归遍历整个树结点，查找遍历数组，最差情况都是O(N)

空间复杂度O(N)：存储二叉树中序遍历数组res占用空间O(N)

 还有一种空间复杂度较小的解法：特性模拟

来自https://blog.nowcoder.net/n/27cc23d47d7247379bd8b8a4dd6a3607?f=comment

先判断目标结点是不是叶子结点？

①如果不是叶子结点，那么在中序遍历中，下一个结点就是当前结点右子树的最左叶子结点。

②如果是叶子结点，如果当前结点是父结点的左子树，那么就要往上寻找父结点。如果当前结点是父节点的右子结点，继续往上找，直到找到当前结点是父结点的左子结点。

class Solution:
    def GetNext(self, pNode):
        # write code here
        if not pNode:
            return pNode
        # 1、如果目标结点不是叶子结点
        while pNode.right:
            p=pNode.right
            # 则在中序遍历中，下一个结点为当前结点的右子树的最左叶子结点
            if p.left:
                p=p.left
            return p
        # 2、目标结点是叶子结点而不是根节点，则需要向上寻找父节点
        while pNode.next:
            p=pNode.next
            # 是父节点的右子结点，则在中序遍历中，还要继续寻找,直到满足当前结点是父节点的左子结点
            if p.left==pNode:
                return p
            pNode=p

时间复杂度 O(N)：最坏的情况是遍历所有结点
空间复杂度 O(1)：没有用到类似数组的类型数据