剑指offer:重建二叉树
admin
2023-07-10 02:23:00
0

题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, preorder, inorder):
        """
        与树有关的题目一般来讲可以先尝试使用递归方法求解。
        :param preorder: 一棵树的先序遍历结果
        :param inorder: 同一棵树的中序遍历结果
        :return: 构造的树的根节点
        """
        def helper(preorder_start, preorder_end, inorder_start, inorder_end):
            """
            通过传下标代替传列表,可以减少中间的内存开销。
            四个参数分别代表了当前子树的先序遍历和中序遍历的结果,根据这四个参数可以从总体的
            先序遍历和中序遍历中找到当前子树的先序遍历和中序遍历。
            :return: 当前子树的根节点
            """
            # 对于一棵子树,其先序遍历的第一个节点就是根节点
            root = TreeNode(preorder[preorder_start])
            # 如果当前子树只有一个节点(这时这个子树其实是叶子节点),这是我们的递归出口
            if preorder_start == preorder_end:
                # 这里增加了一个输入合法性判断
                if inorder_start == inorder_end \
                        and preorder[preorder_end] == inorder[inorder_end]:
                    return root
                else:
                    return None

            # 然后从中序遍历中找到这个根节点,按照中序遍历的顺序,根节点左边的是左子树的节点,
            # 右边的是右子树的节点
            i = inorder_start
            while i <= inorder_end:
                if inorder[i] == root.val:
                    break
                i += 1

            # 输入合法性判断,确保这个根节点在中序遍历中能找到
            if inorder[i] != root.val:
                return None

            left_length = i - inorder_start  # 左子树的节点数
            # 左右子树的先序遍历下标
            preorder_left_start = preorder_start + 1
            preorder_left_end = preorder_left_start + left_length - 1
            preorder_right_start = preorder_left_end + 1
            preorder_right_end = preorder_end

            # 左右子树的中序遍历的下标
            inorder_left_start = inorder_start
            inorder_left_end = i - 1
            inorder_right_start = i + 1
            inorder_right_end = inorder_end

            # 如果存在左右子树,则递归下去
            if left_length > 0:
                root.left = helper(preorder_left_start, preorder_left_end,
                                   inorder_left_start, inorder_left_end)
            if left_length < preorder_end - preorder_start:
                root.right = helper(preorder_right_start, preorder_right_end,
                                    inorder_right_start, inorder_right_end)

            return root

        if not preorder or not inorder:  # 验证输入合法性
            return None

        return helper(0, len(preorder) - 1, 0, len(inorder) - 1)  # 递归构造二叉树

相关内容

热门资讯

美伊谈判濒临破裂之际,伊朗议长... 因为以色列持续对黎巴嫩进行军事打击,伊朗宣布暂停同美国的谈判。不过美国总统特朗普称,对话仍在继续。谈...
罕见!以军政策发生“重大转变” 新华社北京6月1日电 题:罕见纵深推进,以军对黎行动会否搅动美伊谈判新华社记者刘品然 阚静文 席玥以...
山西太原发现一处新石器遗址,出... 山西省太原市文物保护研究院协同相关科研机构,近期在太原市阳曲县西盘威村发现一处新石器时代重要遗址——...
伊媒发布穆杰塔巴罕见照片 伊朗塔斯尼姆通讯社6月1日发布了一张最高领袖穆杰塔巴的照片。照片中,穆杰塔巴面露笑容,抱着一个婴儿。...
福建“泡药杨梅”曝光后,浙江杨... 这两天,浙江本地杨梅少量进入市场。虽然受到此前福建 “泡药杨梅” 事件影响,市场整体销量相比去年同期...
尺素金声 | 前4月规上工业企... 5月27日,国家统计局发布最新数据显示,今年前4月,全国规上工业企业实现利润同比增长18.2%,增速...
郑丽文:台湾民众越来越了解“台... 针对台湾《联合报》民调显示,63%受访者民意希望维持现状,即将访美的中国国民党主席郑丽文1日表示,民...
美前副总统:共和党失去了方向,... 2026年是美国的中期选举年,共和党选情不利,可能在年底的选举中遭遇挫败。美国前副总统彭斯5月31日...
南枝原来去过中国?《给阿嬷的情... 《给阿嬷的情书》票房口碑双丰收,目前票房已突破13亿。凤凰卫视最新一期《问答神州》专访了该片导演蓝鸿...
法国海军扣押一艘俄“影子舰队”... 近日,法国海军在大西洋海域扣押了一艘据称从俄罗斯摩尔曼斯克出发的油轮,引发俄方强烈不满。俄新社6月1...