力扣 105 & 106 从( 前序&&中序 || 中序&&后序 )遍历序列构造二叉树题解

105. 从前序与中序遍历序列构造二叉树

代码解析

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if (preorder.length == 0 || inorder.length == 0) return null;
        //前序遍历结果的第一个数字(preorder[0])是二叉树的根结点
        TreeNode root = new TreeNode(preorder[0]);
        //在中序遍历结果中找到根结点所在位置
        for(int i = 0; i < preorder.length; i++){
            if(preorder[0] == inorder[i]){
            	//中序遍历结果中，位于根结点左边的数字为左子树，位于根结点右边的数字为右子树。
            	//分别提取出左右子树的前序&&中序遍历结果，进行递归
                root.left = buildTree(Arrays.copyOfRange(preorder,1,i+1), Arrays.copyOfRange(inorder,0,i));
                root.right = buildTree(Arrays.copyOfRange(preorder,i+1,preorder.length), Arrays.copyOfRange(inorder,i+1,inorder.length));
                //Arrays.copyOfRange(int[] nums,start,end)提取出数字中索引为start～end的元组(不含end)
                break;
            }
        }
        return root;
    }
}

106. 从中序与后序遍历序列构造二叉树

代码解析

class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        int i_len = inorder.length;
        int p_len = postorder.length;
        if(i_len == 0 || p_len == 0) return null;
        //后序遍历结果的最后一个数字(postorder[p_len-1])是二叉树的根结点
        TreeNode root = new TreeNode(postorder[p_len-1]);
        //在中序遍历结果中找到根结点所在位置
        for(int i = 0; i < p_len; i++){
            if(postorder[p_len-1] == inorder[i]){
            	//中序遍历结果中，位于根结点左边的数字为左子树，位于根结点右边的数字为右子树。
            	//分别提取出左右子树的中序&&后序遍历结果，进行递归
                root.left = buildTree(Arrays.copyOfRange(inorder, 0, i), Arrays.copyOfRange(postorder, 0, i));
                root.right = buildTree(Arrays.copyOfRange(inorder, i+1, i_len), Arrays.copyOfRange(postorder, i, p_len-1));
                break;
            }
        }
        return root;
    }
}