重建二叉树

一、关于Java中的Arrays.copyOfRange()方法

要使用这个方法,首先要import java.util.*;

Arrays.copyOfRange(T[ ] original,int from,int to)

将一个原始的数组original,从小标from开始复制,复制到小标to,生成一个新的数组。

注意这里包括下标from,不包括上标to。

二、重建二叉树

题目来自牛客网:

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

import java.util.*;
/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre.length == 0){
            return null;
        }
        TreeNode node = new TreeNode(pre[0]);
        for(int i = 0;i < in.length; i++){
            if(pre[0] == in[i]){
                node.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
                node.right = reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
            }
        }
        return node;
    }
}

以上是通过递归的方法解决该问题,每一轮的查找,主要是通过前序遍历的root(pre[0]),来中序遍历中寻找到根节点(in[i]),则中序遍历中左侧为左子树,0~i。相对应本次查找的,前序遍历中左子树1~i+1。【根据Arrays.copyOfRange,不包括后续值】同理,对于每一轮的递归,它的右子树前序遍历显然是i+1~pre.length,中序遍历为i+1~in.length.这是一种通过Java中自带函数解决问题的方法,代码简洁易懂。

原文地址:https://www.cnblogs.com/theWinter/p/10521621.html