【剑指Offer】 57.二叉树的下一个结点

题目描述

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

用例

此二叉树的中序遍历的结果为：BKEFACD

解题思路

二叉树(前序，中序，后序，层序)遍历 python实现（递归）

比如给了上面二叉树中的C结点，我们要找中序遍历中C结点下一个结点，首先要找到最顶层的根节点E，然后再通过根节点E，得到二叉树的中序遍历，在中序遍历中找到C结点的下一个结点D。

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

        dummy = pNode
        while dummy.next:
            dummy = dummy.next
        self.result = []
        self.inorderTraversal(dummy)
        if self.result.index(pNode) != len(self.result)-1:
            return self.result[self.result.index(pNode)+1]
        else:
            return None
            
    def inorderTraversal(self, root):  ##中序遍历
		if not root: return
        self.inorderTraversal(root.left)
        self.result.append(root)
        self.inorderTraversal(root.right)

s = Solution()
res = s.searchRange(pNode)
print('.....',res)