[剑指OFFER]之树：JZ7 重建二叉树

最新推荐文章于 2024-07-22 19:36:52 发布

悟空学编程

最新推荐文章于 2024-07-22 19:36:52 发布

阅读量100

点赞数

分类专栏：数据结构与算法文章标签：数据结构算法后端 java 开发语言

本文链接：https://blog.csdn.net/EndlessProcess/article/details/124324101

版权

数据结构与算法专栏收录该内容

42 篇文章 0 订阅

订阅专栏

文章目录

@[TOC]
1.题目
2.解法
2.1 树的结构
2.2 解法

总结

1.题目

给定节点数为 n 的二叉树的前序遍历和中序遍历结果，请重建出该二叉树并返回它的头结点。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建出如下图所示。
在这里插入图片描述

2.解法

2.1 树的结构

 class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;
        public TreeNode(int val) {
            this.val = val;
        }
    }

2.2 解法

前序遍历（或者后序遍历）+ 中序遍历即可恢复这棵树。
题目给出的前+中的方式。
1、我们通过前序遍历的数组就知道第一个元素就是根，于是到中序数组去找根的位置。
2、找到后，根左侧是左子树，右侧是右子树。
3、根据左子树的长度到前序数组去找，左子树。
4、于是又可以得到前序遍历左子树和前序遍历右子树。
5、找到了两组，分别是：前序遍历左子树、中序遍历左子树。前序遍历右子树、中序遍历右子树。
于是，递归构建。
直到得到一个元素的数组或者一个空数组。
在这里插入图片描述

    public TreeNode reConstructBinaryTree(int[] pre, int[] vin) {
        if(pre==null||pre.length==0) return null;

        int rootval = pre[0];

        //查出中序遍历根节点的位置
        int pos = -1;
        for(int i=0;i<vin.length;i++) {
            if(vin[i]==rootval) {
                pos = i;
                break;
            }
        }

        int left_len = pos;
        int right_len = pre.length-1-pos;

        int[] pre_left = new int[left_len];
        int[] pre_right = new int[right_len];
        int[] vin_left = new int[left_len];
        int[] vin_right = new int[right_len];

        System.arraycopy(pre,1,pre_left,0,left_len);
        System.arraycopy(pre,1+left_len,pre_right,0,right_len);
        System.arraycopy(vin,0,vin_left,0,left_len);
        System.arraycopy(vin,1+left_len,vin_right,0,right_len);

        TreeNode node = new TreeNode(rootval);
        node.left = reConstructBinaryTree(pre_left,vin_left);
        node.right = reConstructBinaryTree(pre_right,vin_right);
        return node;
    }