递归重建二叉树的思路

Posted 2021-02-02 tongxupeng

（1）通过前序列表（根左右） 和 中序列表（左跟右）来重建二叉树

       思路 前序遍历 序列中，第一个数字总是二叉树的根节点。在中序遍历 序列中，根节点的值在序列的中间，左子树的节点的值位于根节点的值的左边，右子树的节点的值位于根节点的值的右边。根据二叉树的这个性质，采用递归方法可以重建一个二叉树了。

/**
 * 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) {
        TreeNode node=reConstructBinaryTree(pre,0,pre.length-1,in,0,in.length-1);
        return node;
    }
    public TreeNode reConstructBinaryTree(int [] pre,int ps,int pe,int [] in,int is,int ie) {
        if(ps>pe)//如果开始位置大于结束位置说明已经处理到叶节点了
            return null;
        int value=pre[ps];
        int index=is;
        while(index<=ie&&in[index]!=value)//找到的index为根节点在中序遍历序列中的索引
            index++;
        TreeNode node=new TreeNode(pre[startPre]);
        node.left=reConstructBinaryTree(pre,ps+1,ps+index-is,in,is,index-1);
        node.right=reConstructBinaryTree(pre,ps+index-is+1,pe,in,index+1,ie);
        return node;
    }
}

（2）用Java中的Arrays.copyOfRange完成二叉树的重建工作

Arrays.copyOfRange(T[ ] original,int from,int to) 将一个原始的数组original，从小标from开始复制，复制到小标to，生成一个新的数组。 注意这里包括下标from，不包括下标to。

public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre.length == 0 || in.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;

    }