剑指offer------重建二叉树(java)

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

实现代码:

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
import java.util.*;
public class Solution {
    private Map<Integer, Integer> indexForInOrders = new HashMap<>();
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        for(int i=0;i<in.length;i++){
            indexForInOrders.put(in[i],i);
        }
        return reConstructBinaryTree(pre,0,pre.length-1,0);
    }
    private TreeNode reConstructBinaryTree(int[] pre,int preL,int preR,int inL){
        if(preL>preR) return null;
        TreeNode root = new TreeNode(pre[preL]);
        int inIndex = indexForInOrders.get(root.val);
        int leftTreeNum = inIndex-inL;
        root.left = reConstructBinaryTree(pre,preL+1,preL+leftTreeNum,inL);
        root.right = reConstructBinaryTree(pre,preL+leftTreeNum+1,preR,inL+leftTreeNum+1);
        return root;
    }
}

首先先搞明白前序遍历和中序遍历的原理:
前序遍历:先访问根节点,然后遍历左子树,最后遍历右子树。
中序遍历:先遍历左子树,然后访问根节点,最后遍历右子树。
因此在前序遍历的第一个数字便是根节点,在中序遍历中,该数字的左边是左子树,右边是右子树。左右子树的数列在前序遍历中所对应的第一个数字是该子树的节点,一次类推直到遍历完所有数字。
如果输入输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},那么这棵二叉树如下图
在这里插入图片描述

private Map<Integer, Integer> indexForInOrders = new HashMap<>();
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        for(int i=0;i<in.length;i++){
            indexForInOrders.put(in[i],i);
        }
        return reConstructBinaryTree(pre,0,pre.length-1,0);
    }

这段代码,主要是将中序遍历中数字的顺序存储起来,方便后面递归的时候确定子树在前序遍历中的左右界限。

private TreeNode reConstructBinaryTree(int[] pre,int preL,int preR,int inL){
        if(preL>preR) return null;
        TreeNode root = new TreeNode(pre[preL]);
        int inIndex = indexForInOrders.get(root.val);
        int leftTreeNum = inIndex-inL;
        root.left = reConstructBinaryTree(pre,preL+1,preL+leftTreeNum,inL);
        root.right = reConstructBinaryTree(pre,preL+leftTreeNum+1,preR,inL+leftTreeNum+1);
        return root;
    }

这个方法包含四个参数
(1)前序遍历数组
(2)当前子树在前序遍历中的左界限
(3)当前子树在前序遍历中的右界限
(4)当前子树在中序遍历中的左界限(主要确定当前树的根节点位置)
在这里插入图片描述
执行两条递归语句最终把二叉树重建。

        root.left = reConstructBinaryTree(pre,preL+1,preL+leftTreeNum,inL);
        root.right = reConstructBinaryTree(pre,preL+leftTreeNum+1,preR,inL+leftTreeNum+1);

(方法不唯一!)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值