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

 思路:前序是根左右,前序序列第一个元素一定是根。中序是左,根,右。根节点左边一定是左子树,右边一定是右子树。

树没有重复元素,所以,先找出根节点,初始化一个TreeNode   root,再根据数值相同,找中序遍历里面的根节点,之后用Arrays.copyOfRange(preorder,1,num+1),求出先序遍历,中序遍历下的左右子树的序列,递归调用函数,返回左右子树根节点,最后root连接一下左右子树,返回root就行。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {//看笔记
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder.length>0&&inorder.length>0)
        {
            TreeNode root =new TreeNode(preorder[0]);//根节点

        
        int num=0;//根节点在中序序列里的下标
        for(int i=0;i<inorder.length;i++)
        {
            if(inorder[i]==root.val)
            {
                num=i;//num是下标,左子树加根节点总共num+1个,左子树num个元素
                break;
            }
        }
        int[] preLeft = Arrays.copyOfRange(preorder,1,num+1);//取数组里[a,b)区间的元素,这里是取了num个
        int[] preRight = Arrays.copyOfRange(preorder,num+1,preorder.length);//取出剩下的元素组成的数组,也就是右子树的元素的数组
        int[] inLeft=Arrays.copyOfRange(inorder,0,num);//取出中序遍历里面,左子树的元素组成的数组
        int[] inRight=Arrays.copyOfRange(inorder,num+1,inorder.length);
        root.left=buildTree(preLeft,inLeft);
        root.right=buildTree(preRight,inRight);
        return root;
        }
        else
        {
            return null;
        }
    }
}

  

原文地址:https://www.cnblogs.com/lzh1043060917/p/12835188.html