【剑指Offer】重建二叉树

【剑指Offer】重建二叉树

题目描述

用给定的数组pre[]和in[]来分别表示二叉树的先序遍历序列和后序遍历序列，要求对二叉树进行还原并输出。
此外，二叉树的定义也已经给出（不同的语言采取的定义不同，此处以Java为例）

public class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(int x) { val = x; }
     }

解题思路

此题属于大学本科《数据结构》课程习题的拓展，一般学校考核以手工还原为主。
对于拿到有先序遍历的题来说，我们首先可以确定的是，先序序列的数组第一个元素是根节点。然后再将该节点在中序的数组中的位置找到，那么中序序列两边由此又可分为两棵二叉树。然后根据二叉树的递归定义，递归地在两边进行根节点访问，再插入到树中，即为答案。

代码

代码部分在此笔者采用了Arrays类中的copyOfRange()方法，可以较方便地拷贝数组连续部分（类似于C语言中对于字符串的拷贝）。此外reConstructBinaryTree()方法的定义在题目背景中已经给出，可直接调用。

import java.util.Arrays;
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre.length==0||in.length==0){
        //临界条件
             return null;
        }
            TreeNode resTree=new TreeNode(pre[0]);//先序序列第一个元素是根节点
            for(int i=0;i<in.length;i++){
                if(pre[0]==in[i]){//先序序列队头在中序中找位置
                    resTree.left=reConstructBinaryTree(Arrays.copyOfRange(pre, 1, i + 1), Arrays.copyOfRange(in, 0, i));
                    resTree.right=reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
                }
            }
        return resTree;
    }
}

总结

在不考虑时间复杂度、空间开销的情况下，二叉树问题优先考虑使用递归解法在边界条件下求解。